mirror of https://github.com/gsi-upm/senpy
Added shelve mixin
parent
bd2e0f0d5c
commit
b8993f7d64
@ -0,0 +1,41 @@
|
||||
import os
|
||||
import logging
|
||||
import shelve
|
||||
|
||||
try:
|
||||
import unittest.mock as mock
|
||||
except ImportError:
|
||||
import mock
|
||||
import json
|
||||
import os
|
||||
from unittest import TestCase
|
||||
from senpy.models import Response, Entry
|
||||
from senpy.plugins import SenpyPlugin, ShelfMixin
|
||||
|
||||
|
||||
class ModelsTest(TestCase):
|
||||
|
||||
# def test_shelf(self):
|
||||
# class ShelfTest(ShelfMixin):
|
||||
# pass
|
||||
# a = ShelfTest()
|
||||
# print(type(a.sh))
|
||||
# assert(False)
|
||||
|
||||
def test_shelf(self):
|
||||
class ShelfTest(ShelfMixin, SenpyPlugin):
|
||||
pass
|
||||
a = ShelfTest({'name': 'shelve', 'version': 'test'})
|
||||
print(type(a.sh))
|
||||
a.context = "ohno"
|
||||
del a.context
|
||||
print(a)
|
||||
a.sh['classifier'] = {'name': 'ohno'}
|
||||
assert a.name == 'shelve'
|
||||
assert isinstance(a.sh, shelve.Shelf)
|
||||
a.close()
|
||||
b = ShelfTest({'name': 'shelve', 'version': 'test'})
|
||||
assert b.name == 'shelve'
|
||||
assert b.sh['classifier']
|
||||
assert b.sh['classifier']['name'] == 'ohno'
|
||||
assert isinstance(b.sh, shelve.Shelf)
|
Loading…
Reference in New Issue