def test_get_best_lap_of_race(self):
lap_service = LapService(self.laps)
result = lap_service.get_best_lap_of_race()
self.assertEquals(result.driver.id, self.lap_1.driver.id)
self.assertEquals(result.driver.name, self.lap_1.driver.name)
self.assertEquals(result.number, self.lap_1.number)
self.assertEquals(result.duration, self.lap_1.duration)
def test_get_drivers_average_speed(self):
lap_service = LapService(self.laps)
result = lap_service.get_drivers_average_speed()
self.assertEquals(result[0].id, self.driver_2.id)
self.assertEquals(result[0].name, self.driver_2.name)
self.assertEquals(result[0].average_speed, 44.275)
self.assertEquals(result[1].id, self.driver.id)
self.assertEquals(result[1].name, self.driver.name)
self.assertEquals(result[1].average_speed, 44.275)
def test_get_best_drivers_lap(self):
lap_service = LapService(self.laps)
result = lap_service.get_best_drivers_lap()
self.assertEquals(result[0].driver.id, self.lap_1.driver.id)
self.assertEquals(result[0].driver.name, self.lap_1.driver.name)
self.assertEquals(result[0].number, self.lap_1.number)
self.assertEquals(result[0].duration, self.lap_1.duration)
self.assertEquals(result[1].driver.id, self.lap_2.driver.id)
self.assertEquals(result[1].driver.name, self.lap_2.driver.name)
self.assertEquals(result[1].number, self.lap_2.number)
self.assertEquals(result[1].duration, self.lap_2.duration)
def test_get_last_laps_duration(self):
lap_service = LapService(self.laps)
result = lap_service.get_last_laps_duration()
duration_1 = self.lap_1.duration + self.lap_4.duration
duration_2 = self.lap_2.duration + self.lap_3.duration
self.assertEquals(result[0]['duration'], duration_1)
self.assertEquals(result[0]['lap'], self.lap_4)
self.assertEquals(result[1]['duration'], duration_2)
self.assertEquals(result[1]['lap'], self.lap_3)
def _get_lap_service(self):
try:
lines = self._get_lines_file()
laps = self._get_laps(lines)
except ServiceException as e:
raise ManagerException(e)
return LapService(laps)
def test_get_average_speed_of_laps(self):
lap_service = LapService(self.laps)
result = lap_service._get_average_speed_of_laps(self.laps)
self.assertEquals(result, 44.275)
def test_get_race_duration_of_driver(self):
lap_service = LapService(self.laps)
result = lap_service._get_race_duration_of_driver(
self.drive_1_laps_sorted) # noqa
self.assertEquals(result, self.lap_4.duration + self.lap_1.duration)
def test_get_last_driver_laps(self):
lap_service = LapService(self.laps)
result = lap_service._get_last_driver_laps(self.drive_1_laps_sorted)
self.assertEquals(result, self.lap_4)
def test_get_drivers_id(self):
lap_service = LapService(self.laps)
result = lap_service._get_drivers_id()
self.assertEquals(result, [33, 38])