def testProcessing(self): """ test event 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(28.4) result2 = entities.Result(60.47) result3 = entities.Result(174.3) result1.set_place(8) result2.set_place(5) result3.set_place(10) 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) eventProcessing = processing.EventResults(event1) eventProcessing.process() self.assertEqual(eventProcessing.get_results(), [athlete2, athlete1, athlete3])
def _processing_setup(self) : """ Create a country with athletes and results to test 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(28.4) result2 = entities.Result(60.47) result3 = entities.Result(174.3) result1.set_place(2) result2.set_place(3) result3.set_place(10) 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) country1.add_athlete(athlete1) country1.add_athlete(athlete2) country1.add_athlete(athlete3) event1.add_athlete(athlete1) event1.add_athlete(athlete2) event1.add_athlete(athlete3) return country1
def testAthlete(self): """ test add_athlete and get_athletes""" country1 = entities.Country("Canada", "CAN") athlete1 = entities.Athlete("1635", "John", "Smith", country1) event1 = entities.Event("Skiing", True, []) self.assertEqual(event1.get_athletes(), []) event1.add_athlete(athlete1) self.assertEqual(event1.get_athletes(), [athlete1])
def testEvents(self): """ test get_events and add_event""" country1 = entities.Country("Canada", "CAN") athlete1 = entities.Athlete("12345", "John", "Smith", country1) self.assertEqual(athlete1.get_events(), []) event1 = entities.Event("Skiing", False, []) athlete1.add_event(event1) self.assertEqual(athlete1.get_events(), [event1])
def testResults(self): """ test get_result and add_result""" country1 = entities.Country("Canada", "CAN") athlete1 = entities.Athlete("12345", "John", "Smith", country1) event1 = entities.Event("Skiing", False, []) athlete1.add_event(event1) result1 = entities.Result(15.1) athlete1.add_result(event1, result1) self.assertEqual(athlete1.get_result(event1), result1)
def testProcessing(self): """ test athlete processing """ country1 = entities.Country("Canada", "CAN") athlete1 = entities.Athlete("12345", "John", "Smith", country1) event1 = entities.Event("Skiing 2000m", True, [athlete1]) event2 = entities.Event("Skiing 4000m", True, [athlete1]) event3 = entities.Event("Asking 6000m", True, [athlete1]) athlete1.add_event(event1) athlete1.add_event(event2) athlete1.add_event(event3) result1 = entities.Result(28.4) result2 = entities.Result(60.47) result3 = entities.Result(174.3) result1.set_place(15) result2.set_place(7) result3.set_place(12) athlete1.add_result(event1, result1) athlete1.add_result(event2, result2) athlete1.add_result(event3, result3) athleteProcessing = processing.AthleteResults(athlete1) athleteProcessing.process() self.assertEqual(athleteProcessing.get_results(), [result2, result3, result1])
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()
def testGetResults(self): """ test get_results""" eventResults = processing.EventResults(entities.Event("Skiing", False, [])) with self.assertRaises(ValueError): eventResults.get_results()
def testGetName(self): """ test get_name""" self.assertEqual(entities.Event("Skiing", True, []).get_name(), "Skiing")
def testIsTimed(self): """ test is_timed""" self.assertTrue(entities.Event("Skiing", True, []).is_timed())