def test_nhl_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_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='DET')
type(team)._abbreviation = mock_abbreviation
assert len(team.roster.players) == 2
for player in team.roster.players:
assert player.name in ['Jimmy Howard', 'Henrik Zetterberg']
type(team)._abbreviation = None
def test_team_string_representation(self, *args, **kwargs):
detroit = Team('DET')
assert detroit.__repr__() == 'Detroit Red Wings (DET) - 2017'
def test_pulling_team_directly(self, *args, **kwargs):
detroit = Team('DET')
for attribute, value in self.results.items():
assert getattr(detroit, attribute) == value