import socceraction.atomic.spadl.utils as spu import socceraction.atomic.vaep.labels as lab from pandera.typing import DataFrame from socceraction.atomic.spadl import AtomicSPADLSchema def test_scores(atomic_spadl_actions: DataFrame[AtomicSPADLSchema]) -> None: nr_actions = 10 atomic_spadl_actions = spu.add_names(atomic_spadl_actions) scores = lab.scores(atomic_spadl_actions, nr_actions) assert len(scores) == len(atomic_spadl_actions) def test_conceds(atomic_spadl_actions: DataFrame[AtomicSPADLSchema]) -> None: nr_actions = 10 atomic_spadl_actions = spu.add_names(atomic_spadl_actions) concedes = lab.concedes(atomic_spadl_actions, nr_actions) assert len(concedes) == len(atomic_spadl_actions)