class DataExtractorTest(unittest.TestCase):
def setUp(self):
self.extractor = DataExtractor(raw_data)
self.extractor.extract()
self.race = self.extractor.get_race()
def test_extracts_heat(self):
self.assertEquals(60, self.race.heat)
def test_extracts_race_date_and_time(self):
self.assertEquals(datetime.date(2011,12,23), self.race.date)
self.assertEquals(datetime.time(20,36), self.race.time)
def test_extract_driver_list(self):
drivers = [u'CiglaR', u'CASPER', u'Brzi', u'bR1ck', u'gogoGT', u'Shorty', u'dastrong', u'skrla', u'slavisha', u'VINKO']
self.assertEquals(drivers, self.race.driver_list)
