def test():
team_id = TEAMS['ATL']['id']
player_id = get_player('Lebron', 'James')
assert team.TeamList()
assert team.TeamSummary(team_id)
team_details = team.TeamDetails(team_id)
assert team_details
assert team_details.background()
assert team_details.history()
assert team.TeamCommonRoster(team_id)
assert team.TeamGeneralSplits(team_id)
assert team.TeamOpponentSplits(team_id)
assert team.TeamLastNGamesSplits(team_id)
assert team.TeamInGameSplits(team_id)
assert team.TeamClutchSplits(team_id)
assert team.TeamShootingSplits(team_id)
assert team.TeamPerformanceSplits(team_id)
assert team.TeamLineups(team_id)
assert team.TeamPlayers(team_id)
assert team.TeamPlayerOnOffDetail(team_id)
assert team.TeamPlayerOnOffSummary(team_id)
assert team.TeamGameLogs(team_id)
assert team.TeamShotTracking(team_id)
assert team.TeamReboundTracking(team_id)
assert team.TeamPassTracking(team_id)
assert team.TeamVsPlayer(team_id, player_id)
def testAll(self):
assert team.TeamList()
assert team.TeamSummary(self.teamId)
team_details = team.TeamDetails(self.teamId)
assert team_details
# assert team_details.background()
# assert team_details.history()
assert team.TeamCommonRoster(self.teamId)
assert team.TeamGeneralSplits(self.teamId)
assert team.TeamOpponentSplits(self.teamId)
assert team.TeamLastNGamesSplits(self.teamId)
assert team.TeamInGameSplits(self.teamId)
assert team.TeamClutchSplits(self.teamId)
assert team.TeamShootingSplits(self.teamId)
assert team.TeamPerformanceSplits(self.teamId)
assert team.TeamLineups(self.teamId)
assert team.TeamPlayers(self.teamId)
assert team.TeamPlayerOnOffDetail(self.teamId)
assert team.TeamPlayerOnOffSummary(self.teamId)
assert team.TeamGameLogs(self.teamId)
assert team.TeamShotTracking(self.teamId)
assert team.TeamReboundTracking(self.teamId)
assert team.TeamPassTracking(self.teamId)
assert team.TeamVsPlayer(self.teamId, self.playerId)
VALUE_KEYS = [
u'GP', u'MIN', u'PLUS_MINUS', u'OFF_RATING', u'DEF_RATING', u'NET_RATING'
]
PRIMARY_KEYS = [
u'GROUP_SET', u'TEAM_ID', u'TEAM_ABBREVIATION', u'TEAM_NAME',
u'VS_PLAYER_ID', u'VS_PLAYER_NAME'
]
if __name__ == '__main__':
if os.path.exists(TEAMS_FILE):
teams = pd.DataFrame.from_csv(TEAMS_FILE)
else:
teams = team.TeamList().info().dropna()
teams.to_csv(TEAMS_FILE)
all = None
for index, row in teams.iterrows():
data = team.TeamPlayerOnOffSummary(team_id=row['TEAM_ID'])
on_court = data.on_court().drop([u'COURT_STATUS'], 1)
off_court = data.off_court().drop([u'COURT_STATUS'], 1)
overall = on_court.merge(off_court,
left_on=PRIMARY_KEYS,
right_on=PRIMARY_KEYS,
how='outer',
suffixes=['_ON', '_OFF']).dropna()
ordered_overall = overall[['TEAM_NAME', 'VS_PLAYER_NAME']]
for name in VALUE_KEYS:
keys = []
for suffix in ['_ON', '_OFF']:
full_name = name + suffix
value = overall[[full_name]]