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()