tests/vaep/test_labels.py

import socceraction.spadl.utils as spu
import socceraction.vaep.labels as lab
from pandera.typing import DataFrame
from socceraction.spadl import SPADLSchema


def test_scores(spadl_actions: DataFrame[SPADLSchema]) -> None:
    nr_actions = 10
    spadl_actions = spu.add_names(spadl_actions)
    scores = lab.scores(spadl_actions, nr_actions)
    assert len(scores) == len(spadl_actions)


def test_conceds(spadl_actions: DataFrame[SPADLSchema]) -> None:
    nr_actions = 10
    spadl_actions = spu.add_names(spadl_actions)
    concedes = lab.concedes(spadl_actions, nr_actions)
    assert len(concedes) == len(spadl_actions)