diff --git a/tests/integ/test_tools.py b/tests/integ/test_tools.py index 24bd259f..ba5b989e 100644 --- a/tests/integ/test_tools.py +++ b/tests/integ/test_tools.py @@ -22,6 +22,7 @@ grounding_sam, ixc25_image_vqa, ixc25_video_vqa, + ixc25_temporal_localization, loca_visual_prompt_counting, loca_zero_shot_counting, ocr, @@ -238,6 +239,17 @@ def test_ixc25_video_vqa() -> None: assert "cat" in result.strip() +def test_ixc25_temporal_localization() -> None: + frames = [ + np.array(Image.fromarray(ski.data.cat()).convert("RGB")) for _ in range(10) + ] + result = ixc25_temporal_localization( + prompt="What animal is in this video?", + frames=frames, + ) + assert result == [True] * 10 + + def test_ocr() -> None: img = ski.data.page() result = ocr(