def test_should_add_event(self): # Arrange & Act state = FootballSimulationGameState() \ .event(Event('my event 1')) \ .event(Event('second event')) # Assert self.assertEqual(state.get_events()[0].get_description(), 'my event 1') self.assertEqual(state.get_events()[1].get_description(), 'second event')
def test_should_only_occur_once_a_quarter(self): # Arrange state = FootballSimulationGameState() \ .team1(FootballTeam('teama')) \ .team2(FootballTeam('teamb')) \ .quarter(2) \ .time(0) # Act state = quarter_banner(state) state = quarter_banner(state) # Assert types = [type(t) for t in state.get_events()] self.assertEqual(1, types.count(QuarterStartBannerEvent))
def test_get_events_should_be_zero_length_at_constructor(self): # Arrange state = FootballSimulationGameState() # Assert self.assertEqual(len(state.get_events()), 0)