def test_set_state_in_memory_resource(self): handler = CSVFile(string=TEST_DATA_2) handler.add_row(['a', 'b', 'c']) data = handler.to_str() handler2 = CSVFile(string=data) self.assertEqual(handler2.get_row(3), ['a', 'b', 'c'])
def test_set_state_in_file_resource(self): handler = CSVFile('tests/test.csv') handler.add_row(['d1', 'e1', 'f1']) handler.save_state() handler2 = CSVFile('tests/test.csv') self.assertEqual(handler2.get_row(3), ['d1', 'e1', 'f1']) handler2.del_row(3) handler2.save_state() handler = CSVFile('tests/test.csv') self.assertEqual(handler.get_nrows(), 3)
def test_add_row(self): handler = CSVFile(string=TEST_DATA_2) handler.add_row(['a', 'b', 'c']) self.assertEqual(handler.get_row(3), ['a', 'b', 'c'])
def test_get_row(self): handler = CSVFile(string=TEST_DATA_2) self.assertEqual(handler.get_row(1), ['four', 'five', 'six'])