| import os |
| from datetime import datetime |
|
|
| import pandas as pd |
| from pytest import fixture |
| from socceraction.data.opta import ( |
| OptaCompetitionSchema, |
| OptaGameSchema, |
| OptaPlayerSchema, |
| OptaTeamSchema, |
| ) |
| from socceraction.data.opta.parsers import MA1JSONParser |
|
|
|
|
| @fixture() |
| def ma1json_parser() -> MA1JSONParser: |
| path = os.path.join( |
| os.path.dirname(__file__), |
| os.pardir, |
| os.pardir, |
| os.pardir, |
| "datasets", |
| "opta", |
| "ma1_408bfjw6uz5k19zk4am50ykmh.json", |
| ) |
| return MA1JSONParser(str(path)) |
|
|
|
|
| def test_extract_competitions(ma1json_parser: MA1JSONParser) -> None: |
| competitions = ma1json_parser.extract_competitions() |
| assert len(competitions) == 1 |
| assert competitions[("722fdbecxzcq9788l6jqclzlw", "408bfjw6uz5k19zk4am50ykmh")] == { |
| "competition_id": "722fdbecxzcq9788l6jqclzlw", |
| "season_id": "408bfjw6uz5k19zk4am50ykmh", |
| "competition_name": "2. Bundesliga", |
| "season_name": "2015/2016", |
| } |
| OptaCompetitionSchema.validate(pd.DataFrame.from_dict(competitions, orient="index")) |
|
|
|
|
| def test_extract_games(ma1json_parser: MA1JSONParser) -> None: |
| games = ma1json_parser.extract_games() |
| assert len(games) == 1 |
| assert games["bsu6pjne1eqz2hs8r3685vbhl"] == { |
| "game_id": "bsu6pjne1eqz2hs8r3685vbhl", |
| "season_id": "408bfjw6uz5k19zk4am50ykmh", |
| "competition_id": "722fdbecxzcq9788l6jqclzlw", |
| "game_day": 22, |
| "game_date": datetime(2016, 2, 20, 12, 0), |
| "home_team_id": "aojwbjr39s1w2mcd9l2bf2dhk", |
| "away_team_id": "kxpw3rqn4ukt7nqmtjj62lbn", |
| "venue": "BBBank Wildpark", |
| "away_score": 2, |
| "home_score": 2, |
| "duration": 93, |
| "attendance": 12746, |
| "referee": "Robert Kampka", |
| } |
| OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index")) |
|
|
|
|
| def test_extract_teams(ma1json_parser: MA1JSONParser) -> None: |
| teams = ma1json_parser.extract_teams() |
| assert len(teams) == 2 |
| assert teams["aojwbjr39s1w2mcd9l2bf2dhk"] == { |
| "team_id": "aojwbjr39s1w2mcd9l2bf2dhk", |
| "team_name": "Karlsruher SC", |
| } |
| OptaTeamSchema.validate(pd.DataFrame.from_dict(teams, orient="index")) |
|
|
|
|
| def test_extract_players(ma1json_parser: MA1JSONParser) -> None: |
| players = ma1json_parser.extract_players() |
| assert len(players) == 36 |
| assert players[("bsu6pjne1eqz2hs8r3685vbhl", "b40xhpgxf8cvruo6vumzu3u1h")] == { |
| "game_id": "bsu6pjne1eqz2hs8r3685vbhl", |
| "player_id": "b40xhpgxf8cvruo6vumzu3u1h", |
| "player_name": "Enrico Valentini", |
| "team_id": "aojwbjr39s1w2mcd9l2bf2dhk", |
| "jersey_number": 22, |
| "minutes_played": 93, |
| "starting_position": "Defender", |
| "is_starter": True, |
| } |
| |
| assert ( |
| players[("bsu6pjne1eqz2hs8r3685vbhl", "49797zk0b4wmp4tevwmaeeh91")]["minutes_played"] == 57 |
| ) |
| |
| assert players[("bsu6pjne1eqz2hs8r3685vbhl", "yuw4a34cpasw5e4vqsg6ex1x")][ |
| "minutes_played" |
| ] == (93 - 57) |
| |
| assert ( |
| players[("bsu6pjne1eqz2hs8r3685vbhl", "2175hvbfk4jn4lnj3cetfpp1")]["minutes_played"] == 60 |
| ) |
| OptaPlayerSchema.validate(pd.DataFrame.from_dict(players, orient="index")) |
|
|
