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