def testAthletes(self): """ test add_athlete and get_athletes""" country1 = entities.Country("Canada", "CAN") ath1 = entities.Athlete("1635", "John", "Smith", country1) self.assertEqual(country1.get_athletes(), []) country1.add_athlete(ath1) self.assertEqual(country1.get_athletes(), [ath1])
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 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 testGetResults(self): """ test get_results""" athleteResults = processing.AthleteResults( entities.Athlete("12345", "John", "Smith", entities.Country("Canada", "CAN"))) with self.assertRaises(ValueError): athleteResults.get_results()
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 fails before processing is called. """ countryResults = processing.CountryResults(entities.Country("Canada", "CAN")) with self.assertRaises(ValueError): countryResults.get_results()
def testCountry(self): """ test get_country""" country1 = entities.Country("Canada", "CAN") self.assertEqual(entities.Athlete("12345", "John", "Smith", country1).get_country(), country1)
def testName(self): """ test get_full_name""" country1 = entities.Country("Canada", "CAN") self.assertEqual(entities.Athlete("12345", "John", "Smith", country1).get_full_name(), "John Smith")
def testGetCountryCode(self): """ test get_country_code""" self.assertEqual(entities.Country("Australia", "AUS").get_country_code(), "AUS")
def testGetName(self): """ test get_name""" self.assertEqual(entities.Country("Canada", "CAN").get_name(), "Canada")