def test_nfl_schedule_returns_schedule(self, *args, **kwargs): flexmock(Team) \ .should_receive('_parse_team_data') \ .and_return(None) flexmock(Schedule) \ .should_receive('_pull_schedule') \ .and_return(None) team = Team(None, 1) assert len(team.schedule) == 0
def test_pulling_team_directly(self, *args, **kwargs): schedule = MockSchedule(None, None) flexmock(Team) \ .should_receive('schedule') \ .and_return(schedule) kansas = Team('KAN') for attribute, value in self.results.items(): assert getattr(kansas, attribute) == value
def test_roster_from_team_class(self, *args, **kwargs): flexmock(Team) \ .should_receive('_parse_team_data') \ .and_return(None) team = Team(None, 1, '2018') mock_abbreviation = mock.PropertyMock(return_value='NOR') type(team)._abbreviation = mock_abbreviation assert len(team.roster.players) == 5 for player in team.roster.players: assert player.name in ['Drew Brees', 'Demario Davis', 'Tommylee Lewis', 'Wil Lutz', 'Thomas Morstead'] type(team)._abbreviation = None
def test_team_string_representation(self, *args, **kwargs): kansas = Team('KAN') assert kansas.__repr__() == 'Kansas City Chiefs (KAN) - 2017'