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