class PickleTourListWriteTestCase(unittest.TestCase): """Test Writer part""" def setUp(self): self.tourListIO = PickleTourListIO() def testWrite(self): """test if we can write a list of integers correctly""" tours = [1,2,3] # it does not matter that this is not a real tour list self.tourListIO.write_tour_list(tours, TEST_DATA_FILE) # verify contents data_file = file(TEST_DATA_FILE, 'rb') tours_read = cPickle.load(data_file) data_file.close() os.remove(TEST_DATA_FILE) self.assertEqual(tours, tours_read) def testWriteRealTours(self): from importer.HAC4FileImporter import HAC4FileImporter file_importer = HAC4FileImporter(IMPORT_FILE) from model.HAC4TourList import HAC4TourList tourList = HAC4TourList(file_importer.doImport()) self.tourListIO.write_tour_list(tourList, TEST_DATA_FILE) #verify data_file = file(TEST_DATA_FILE, 'rb') tourListRead = cPickle.load(data_file) data_file.close() os.remove(TEST_DATA_FILE) self.assertEqual(tourList.get_tours(), tourListRead.get_tours())
class PickleTourListWriteTestCase(unittest.TestCase): """Test Writer part""" def setUp(self): self.tourListIO = PickleTourListIO() def testWrite(self): """test if we can write a list of integers correctly""" tours = [1, 2, 3] # it does not matter that this is not a real tour list self.tourListIO.write_tour_list(tours, TEST_DATA_FILE) # verify contents data_file = file(TEST_DATA_FILE, 'rb') tours_read = cPickle.load(data_file) data_file.close() os.remove(TEST_DATA_FILE) self.assertEqual(tours, tours_read) def testWriteRealTours(self): from importer.HAC4FileImporter import HAC4FileImporter file_importer = HAC4FileImporter(IMPORT_FILE) from model.HAC4TourList import HAC4TourList tourList = HAC4TourList(file_importer.doImport()) self.tourListIO.write_tour_list(tourList, TEST_DATA_FILE) #verify data_file = file(TEST_DATA_FILE, 'rb') tourListRead = cPickle.load(data_file) data_file.close() os.remove(TEST_DATA_FILE) self.assertEqual(tourList.get_tours(), tourListRead.get_tours())