File size: 1,196 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 |
"""Configuration for pytest."""
import os
from collections.abc import Iterator
import pandas as pd
import pytest
from _pytest.config import Config
from pandera.typing import DataFrame
from socceraction.atomic.spadl import AtomicSPADLSchema
from socceraction.spadl import SPADLSchema
def pytest_configure(config: Config) -> None:
"""Pytest configuration hook."""
config.addinivalue_line("markers", "e2e: mark as end-to-end test.")
@pytest.fixture(scope="session")
def sb_worldcup_data() -> Iterator[pd.HDFStore]:
hdf_file = os.path.join(
os.path.dirname(__file__), "datasets", "statsbomb", "spadl-WorldCup-2018.h5"
)
store = pd.HDFStore(hdf_file, mode="r")
yield store
store.close()
@pytest.fixture(scope="session")
def spadl_actions() -> DataFrame[SPADLSchema]:
json_file = os.path.join(os.path.dirname(__file__), "datasets", "spadl", "spadl.json")
return pd.read_json(json_file, orient="records")
@pytest.fixture(scope="session")
def atomic_spadl_actions() -> DataFrame[AtomicSPADLSchema]:
json_file = os.path.join(os.path.dirname(__file__), "datasets", "spadl", "atomic_spadl.json")
return pd.read_json(json_file, orient="records")
|