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