def test_add_kickoff_receiving_play_should_be_able_to_add(self): # Arrange play = Mock(spec=KickOffReceivingPlay) # Act playbook = FootballPlaybook() \ .add_kickoff_receiving_play(play) # Assert self.assertTrue(play in playbook.get_kickoff_receiving_plays())
def test_remove_kickoff_receiving_play_be_able_to_remove(self): # Arrange play = Mock(spec=KickOffReceivingPlay) # Act playbook = FootballPlaybook() \ .add_kickoff_receiving_play(play) \ .remove_kickoff_receiving_play(play) # Assert self.assertFalse(play in playbook.get_kickoff_receiving_plays())