File size: 1,796 Bytes
d6ea71e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
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)
|