def get_outcome(self, outcome_name): """Query a constructed wheel for getting outcome.""" # create Whell wheel = Wheel() return wheel.get_outcome(outcome_name)
def test_get_outcome(): wheel = Wheel() outcome_5 = Outcome("Five", 6) wheel.add_outcome(4, outcome_5) assert wheel.get_outcome("Five") == outcome_5