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