예제 #1
0
    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())
예제 #2
0
    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())