add test case

landing-ai · Sep 11, 2024 · a3f0c35 · a3f0c35
1 parent 4837177
commit a3f0c35
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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(