def test_get_cells_and_data(self):
cells = [CellID('N01'), CellID('N02'), CellID('N03')]
data = Data('test')
cell_data_set = {
cells[0].value: (cells[0], data),
cells[1].value: (cells[1], data),
cells[2].value: (cells[2], data)
}
c_dataset = CellDataSet('id', cell_data_set=cell_data_set)
cells_data = c_dataset.get_cells_and_data()
self.assertEqual(cells_data, [(cells[0], data), (cells[1], data),
(cells[2], data)])
def test_update_cell_in_cell_datasets(self):
store.dropAll()
cds = CellDataSet("id")
cells = ['P0', 'P1', 'P2', 'P3', 'P4', 'P5']
for cell in cells:
cds.add(CellID(cell), Data(""))
store.insert(cds)
store.update_cell_in_cell_datasets("id", CellID('P0'), Data("test"))
stored_cds = store.query_by_cell_in_cell_datasets("id", CellID('P0'))
num_cds = 0
num_cells = 0
for cds in stored_cds:
for cell, data in cds.get_cells_and_data():
assert cells.__contains__(cell.value)
assert data.content == Data("test").content
num_cells = num_cells + 1
num_cds = num_cds + 1
self.assertEqual(num_cds, 1)
self.assertEqual(num_cells, 1)
store.dropAll()
