| import os |
| from datetime import datetime |
|
|
| import pandas as pd |
| from pytest import fixture |
| from socceraction.data.opta import OptaEventSchema, OptaGameSchema |
| from socceraction.data.opta.parsers import F24JSONParser |
|
|
|
|
| @fixture() |
| def f24json_parser() -> F24JSONParser: |
| path = os.path.join( |
| os.path.dirname(__file__), |
| os.pardir, |
| os.pardir, |
| os.pardir, |
| "datasets", |
| "opta", |
| "match-2017-8-918893.json", |
| ) |
| return F24JSONParser(str(path)) |
|
|
|
|
| def test_extract_games(f24json_parser: F24JSONParser) -> None: |
| games = f24json_parser.extract_games() |
| assert len(games) == 1 |
| assert games[918893] == { |
| "game_id": 918893, |
| "season_id": 2017, |
| "competition_id": 8, |
| "game_day": 1, |
| "game_date": datetime(2017, 8, 11, 18, 45, 0, 0), |
| "home_team_id": 3, |
| "away_team_id": 13, |
| } |
| OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index")) |
|
|
|
|
| def test_extract_events(f24json_parser: F24JSONParser) -> None: |
| events = f24json_parser.extract_events() |
| assert len(events) == 1785 |
| assert events[(918893, 1815408644)] == { |
| "game_id": 918893, |
| "event_id": 1815408644, |
| "period_id": 2, |
| "team_id": 3, |
| "player_id": 41792, |
| "type_id": 5, |
| "timestamp": datetime(2017, 8, 11, 20, 38, 49, 0), |
| "minute": 94, |
| "second": 57, |
| "outcome": False, |
| "start_x": 101.1, |
| "start_y": 44.4, |
| "end_x": 101.1, |
| "end_y": 44.4, |
| "qualifiers": { |
| 233: "690", |
| 56: "Center", |
| }, |
| "assist": False, |
| "keypass": False, |
| } |
| df = pd.DataFrame.from_dict(events, orient="index") |
| df["type_name"] = "Added later" |
| OptaEventSchema.validate(df) |
|
|
