| 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) | |