File size: 627 Bytes
d6ea71e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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)
|