def test_add_offense_play_should_be_able_to_add(self): # Arrange play = Mock(spec=OffensePlay) # Act playbook = FootballPlaybook() \ .add_offense_play(play) # Assert self.assertTrue(play in playbook.get_offense_plays())
def test_remove_offense_play_be_able_to_remove(self): # Arrange play = Mock(spec=OffensePlay) # Act playbook = FootballPlaybook() \ .add_offense_play(play) \ .remove_offense_play(play) # Assert self.assertFalse(play in playbook.get_offense_plays())