def test_teams(): test_data = get_test_data('test_teams') for replay_id, ref_teams in test_data.items(): replay_data = get_replay_data(file(get_replay_path(replay_id))) game_data = get_game_data(replay_data) teams = get_teams(game_data) print teams print ref_teams print (sorted(teams[TEAM_1])==sorted(ref_teams[TEAM_1]) and sorted(teams[TEAM_2])==sorted(ref_teams[TEAM_2]))
def dump_teams(): replay_teams = {} for replay_id in get_replay_ids(): try: replay_data = get_replay_data(file(get_replay_path(replay_id))) game_data = get_game_data(replay_data) if 'shuffle_players' in game_data['modes']: replay_teams[replay_id] = get_teams(game_data) except KeyboardInterrupt: raise except: traceback.print_exc() pass pprint(replay_teams)