|
|
import os |
|
|
from datetime import datetime |
|
|
|
|
|
import pandas as pd |
|
|
from pytest import fixture |
|
|
from socceraction.data.opta import OptaCompetitionSchema, OptaGameSchema |
|
|
from socceraction.data.opta.parsers import F1JSONParser |
|
|
|
|
|
|
|
|
@fixture() |
|
|
def f1json_parser() -> F1JSONParser: |
|
|
path = os.path.join( |
|
|
os.path.dirname(__file__), |
|
|
os.pardir, |
|
|
os.pardir, |
|
|
os.pardir, |
|
|
"datasets", |
|
|
"opta", |
|
|
"tournament-2017-8.json", |
|
|
) |
|
|
return F1JSONParser(str(path)) |
|
|
|
|
|
|
|
|
def test_extract_competitions(f1json_parser: F1JSONParser) -> None: |
|
|
competitions = f1json_parser.extract_competitions() |
|
|
assert len(competitions) == 1 |
|
|
assert competitions[(8, 2017)] == { |
|
|
"competition_id": 8, |
|
|
"season_id": 2017, |
|
|
"competition_name": "English Premier League", |
|
|
"season_name": "2017", |
|
|
} |
|
|
OptaCompetitionSchema.validate(pd.DataFrame.from_dict(competitions, orient="index")) |
|
|
|
|
|
|
|
|
def test_extract_games(f1json_parser: F1JSONParser) -> None: |
|
|
games = f1json_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, 19, 45), |
|
|
"home_team_id": 3, |
|
|
"away_team_id": 13, |
|
|
"home_score": 4, |
|
|
"away_score": 3, |
|
|
} |
|
|
OptaGameSchema.validate(pd.DataFrame.from_dict(games, orient="index")) |
|
|
|
