def get_standings(self, standings):
if not standings:
raise TypeError("Wrong argument in get_standings method")
result = []
teams = NBATeams(self.server)
for item in standings:
tmp = dict([('win', item['win']), ('tricode', teams.get_team_tricode_from_id(int(item['teamId']))),
('loss', item['loss'])])
result.append(tmp)
return result
def get_player(self, name, surname, mode):
player_id = None
for player in self.players_list:
if player['firstName'] == name and player['lastName'] == surname:
player_id = player['personId']
if not player_id:
raise NoDataFoundError
player_info = self.server.get_player_stats(player_id)
pi = player_info['stats'][mode]
teams = NBATeams(self.server)
return Player(
name, surname,
teams.get_team_tricode_from_id(int(player_info['teamId'])),
pi['ppg'], pi['rpg'], pi['apg'], pi['mpg'], pi['spg'], pi['bpg'])
def test_get_team_tricode_from_id_correct_id(self):
if NBATeams(server).get_team_tricode_from_id(1610612760) == 'OKC':
pass
def test_get_team_correct_tricode(self):
self.assertIsNotNone(NBATeams(server).get_team("OKC"))
def test_get_team_correct_id(self):
self.assertIsNotNone(NBATeams(server).get_team(1610612760))
def test_get_team_corect_name(self):
self.assertIsNotNone(NBATeams(server).get_team("Chicago Bulls"))
def test_get_team_tricode_from_id_wrong_id_type(self):
try:
NBATeams(server).get_team_tricode_from_id("Type")
except AttributeError:
pass
def test_get_team_tricode_from_id_wrong_id(self):
try:
NBATeams(server).get_team_tricode_from_id(1)
except NameError:
pass
def test_get_team_wrong_tricode(self):
try:
NBATeams(server).get_team("ABC")
except NameError:
pass
def test_get_team_wrong_id(self):
try:
NBATeams(server).get_team(1)
except NameError:
pass
def test_get_team_wrong_name(self):
try:
NBATeams(server).get_team("Team")
except NameError:
pass
def test_init_correct_server(self):
self.assertIsNotNone(NBATeams(server))