def testProcessing(self): """ test data processing from loaded file""" for event in entities.all_events.get_items(): processing.DeterminePlaces(event).process() countryProcessing = processing.CountryResults( entities.all_countries.find_item("GER")) countryProcessing.process() self.assertEqual(countryProcessing.get_results(), [1, 1, 1, 3]) eventProcessing = processing.EventResults( entities.all_events.find_item("Men's Speedskating 5000m")) eventProcessing.process() self.assertEqual(eventProcessing.get_results(), [ entities.all_athletes.find_item("60"), entities.all_athletes.find_item("54"), entities.all_athletes.find_item("61") ]) athlete50 = entities.all_athletes.find_item("50") athleteProcessing = processing.AthleteResults(athlete50) athleteProcessing.process() self.assertEqual(athleteProcessing.get_results(), [ athlete50.get_result( entities.all_events.find_item("Men's Speedskating 500m")), athlete50.get_result( entities.all_events.find_item("Men's Speedskating 1000m")) ])
def testProcessing(self): """ test place processing """ country1 = entities.Country("Canada", "CAN") athlete1 = entities.Athlete("12345", "John", "Smith", country1) athlete2 = entities.Athlete("12346", "David", "Smith", country1) athlete3 = entities.Athlete("12347", "Thomas", "Jones", country1) event1 = entities.Event("Skiing 2000m", True, []) result1 = entities.Result(128.4) result2 = entities.Result(60.47) result3 = entities.Result(174.3) athlete1.add_event(event1) athlete2.add_event(event1) athlete3.add_event(event1) athlete1.add_result(event1, result1) athlete2.add_result(event1, result2) athlete3.add_result(event1, result3) event1.add_athlete(athlete1) event1.add_athlete(athlete2) event1.add_athlete(athlete3) places = processing.DeterminePlaces(event1) places.process() self.assertEqual(places.get_results(), [athlete2, athlete1, athlete3]) self.assertEqual(result1.get_place(), "2") self.assertEqual(result2.get_place(), "1")
def testGetResults(self): """ test get_results""" placeResults = processing.DeterminePlaces(entities.Event("Skiing 2000m", True, [])) with self.assertRaises(ValueError): placeResults.get_results()