class PickleTourListReadTestCase(unittest.TestCase): """Test Reader part""" def setUp(self): self.tourListIO = PickleTourListIO() def testRead(self): """test if we can read a list of integers correctly""" tours = [1,2,3] # write tours to a file data_file = file(TEST_DATA_FILE, 'wb') cPickle.dump(tours, data_file) data_file.close() # verify tours_read = self.tourListIO.read_tour_list(TEST_DATA_FILE) os.remove(TEST_DATA_FILE) self.assertEqual(tours, tours_read) def testWriteTourList(self): from importer.HAC4FileImporter import HAC4FileImporter file_importer = HAC4FileImporter(IMPORT_FILE) from model.HAC4TourList import HAC4TourList tourList = HAC4TourList(file_importer.doImport()) # write tours to a file data_file = file(TEST_DATA_FILE, 'wb') cPickle.dump(tourList, data_file) data_file.close() tourListRead = self.tourListIO.read_tour_list(TEST_DATA_FILE) os.remove(TEST_DATA_FILE) self.assertEqual(tourList.get_tours(), tourListRead.get_tours())
class PickleTourListReadTestCase(unittest.TestCase): """Test Reader part""" def setUp(self): self.tourListIO = PickleTourListIO() def testRead(self): """test if we can read a list of integers correctly""" tours = [1, 2, 3] # write tours to a file data_file = file(TEST_DATA_FILE, 'wb') cPickle.dump(tours, data_file) data_file.close() # verify tours_read = self.tourListIO.read_tour_list(TEST_DATA_FILE) os.remove(TEST_DATA_FILE) self.assertEqual(tours, tours_read) def testWriteTourList(self): from importer.HAC4FileImporter import HAC4FileImporter file_importer = HAC4FileImporter(IMPORT_FILE) from model.HAC4TourList import HAC4TourList tourList = HAC4TourList(file_importer.doImport()) # write tours to a file data_file = file(TEST_DATA_FILE, 'wb') cPickle.dump(tourList, data_file) data_file.close() tourListRead = self.tourListIO.read_tour_list(TEST_DATA_FILE) os.remove(TEST_DATA_FILE) self.assertEqual(tourList.get_tours(), tourListRead.get_tours())