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)