Exemple #1
0
 def test_find_all(self, dataset: EventDataset):
     """
     Test find_all allows simple 'css selector' (<event_type>.<result>)
     """
     goals = dataset.find_all("shot.goal")
     assert len(goals) == 3
     assert goals[0].prev("shot.goal") is None
     assert goals[0].next("shot.goal") == goals[1]
     assert goals[0].next("shot.goal") == goals[2].prev("shot.goal")
     assert goals[2].next("shot.goal") is None
Exemple #2
0
    def test_navigation(self, dataset: EventDataset):
        """
        Test navigating (next/prev) through events
        """
        passes = dataset.find_all("pass")
        assert passes[0].next("pass") == passes[1]
        assert passes[1].prev("pass") == passes[0]
        assert passes[0].next() == dataset.get_event_by_id(
            "61da36dc-d862-416c-8ee3-1a0cd24dc086"
        )

        goals = dataset.find_all("shot.goal")
        assert len(goals) == 3
        assert goals[0].prev("shot.goal") is None
        assert goals[0].next("shot.goal") == goals[1]
        assert goals[0].next("shot.goal") == goals[2].prev("shot.goal")
        assert goals[2].next("shot.goal") is None

        first_goal = dataset.find("shot.goal")
        assert first_goal == goals[0]
        assert first_goal.next(".goal") == goals[1]
        assert first_goal.next(".goal").next(".goal") == goals[2]
        assert first_goal.next(".goal").next(".goal").next(".goal") is None