def setUp(self):
self.df = pn.read_csv(RSC_DIR + '/census.csv')
with open(RSC_DIR + '/schema_census') as f:
schema = json.loads(f.read())
self.schema = OrderedDict(attributes=schema['attributes'],
index=schema['index'])
self.callback_executed = False
self.dispatcher = RPCDispatcher()
self.service = TableService()
self.service.register_in(self.dispatcher)
# Create a table
data = self.df
self._census = self.service.new_table('_census', data, self.schema)
self._census_oid = self._census.oid
class Test(unittest.TestCase):
def callback(self, topic, msg):
print topic, msg
self.callback_executed = True
def setUp(self):
self.df = pn.read_csv(RSC_DIR + '/census.csv')
with open(RSC_DIR + '/schema_census') as f:
schema = json.loads(f.read())
self.schema = OrderedDict(attributes=schema['attributes'],
index=schema['index'])
self.callback_executed = False
self.dispatcher = RPCDispatcher()
self.service = TableService()
self.service.register_in(self.dispatcher)
# Create a table
data = self.df
self._census = self.service.new_table('_census', data, self.schema)
self._census_oid = self._census.oid
def tearDown(self):
names.clear()
def testCreationAsList(self):
data = []
for i in range(len(self.df)):
data.append(self.df.ix[i].to_dict())
self.service.new_table('census', data, self.schema)
request = JSONRPCRequest()
request.method = 'new_table'
request.args = 'census2', data, self.schema
self.dispatcher.dispatch(request)
def testCreationAsDataFrame(self):
data = self.df
self.service.new_table('census', data, self.schema)
request = JSONRPCRequest()
request.method = 'new_table'
request.args = 'census2', data, self.schema
self.dispatcher.dispatch(request)
def testGetData(self):
res = self.service.get_data(self._census_oid)
self.assertEqual(len(res), len(self.df))
request = JSONRPCRequest()
request.method = 'get_data'
request.args = [self._census_oid]
request.unique_id = 1
res = self.dispatcher.dispatch(request).result
self.assertEqual(len(res), len(self.df))
def testFindOne(self):
result = self.service.find_one(self._census_oid, {'$or':[{'State': 'NY'},{'State': 'DC'}]})
self.assertIsInstance(result, dict)
self.assertIn(result['State'], ['DC','NY'])
def testFind(self):
view = self.service.find(self._census_oid, {'$or':[{'State': 'NY'},{'State': 'DC'}]})
self.assertIsInstance(view, TableView)
for result in view.get_data():
self.assertIn(result['State'], ['DC','NY'])
def testRowCount(self):
self.assertEqual(self.service.row_count(self._census_oid), 51)
view = self.service.find(self._census_oid,{'$or':[{'State': 'NY'},{'State': 'DC'}]})
self.assertEqual(view.row_count(), 2)
def testColumnCount(self):
self.assertEqual(self.service.column_count(self._census_oid), 22)
view = self.service.find(self._census_oid, {}, {'Information':True})
self.assertEqual(view.column_count(), 1)
def testColumnNames(self):
self.assertEqual(len(self.service.column_names(self._census_oid)), 22)
view = self.service.find(self._census_oid, {}, {'Information':True})
self.assertEqual(view.column_names(), ['Information'])
def testCheckIndex(self):
with self.assertRaises(ValueError):
self.service._check_index(self._census_oid, {'life_meaning':42})
self.service._check_index(self._census_oid, {'State': 'ES', 'life_meaning':42})
def testIndex(self):
self.assertEqual(self.service.index(self._census_oid), 'State')
def testState(self):
self.assertIsInstance(self.service.schema(self._census_oid), TableSchema)
def testInsert(self):
c1 = self.service.row_count(self._census_oid)
self.service.insert(self._census_oid, {'State': 'ES', 'life_meaning':42})
self.assertEqual(self.service.row_count(self._census_oid) - c1, 1)
c2 = self.service.row_count(self._census_oid)
self.service.insert(self._census_oid,
[{'State': 'ES', 'life_meaning':42},
{'State': 'ES2', 'life_meaning':42},])
self.assertEqual(self.service.row_count(self._census_oid) - c2, 2)
view = self.service.find(self._census_oid, {'life_meaning': {'$exists':True}})
self.assertEqual(view.row_count(), 3)
self.assertEqual( self.service.find_one(self._census_oid, {'life_meaning': {'$exists':True}})['life_meaning'], 42)
def testAddEvent(self):
table = Table('census', self.schema).data(self.df)
table.subscribe_once('add', self.callback)
self.callback_executed = False
table.insert({'State': 'ES', 'life_meaning':42})
self.assertTrue(self.callback_executed)
def testUpdate(self):
val = self.service.find_one(self._census_oid, {'State': 'DC'}, {'Information':True})['Information']
val -= 2000
self.service.update(self._census_oid, {'State': 'DC'}, {'$set': {'Information':val}})
self.assertEqual(self.service.find_one(self._census_oid, {'State': 'DC'}, {'Information':True})['Information'], val)
def testUpdateEvent(self):
# TODO: ¿Do the table service provide subscribing facilities
table = self._census
table.subscribe_once('update', self.callback)
self.callback_executed = False
table.update({'State': 'DC'}, {'$set': {'Information':2000}})
self.assertTrue(self.callback_executed)
def testRemove(self):
query = {'State': 'DC'}
c1 = self.service.find(self._census_oid, query).row_count()
self.service.remove(self._census_oid, query)
self.assertGreater(c1, self.service.find(self._census_oid, query).row_count())
def testRemoveEvent(self):
# TODO: Do the table service provide subscribing facilities?
query = {'State': 'DC'}
self._census.subscribe_once('remove', self.callback)
self.callback_executed = False
self._census.remove(query)
self.assertTrue(self.callback_executed)