From d8b8b7186678727231a0b7fbffc2ef6c2ccfa43a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 19 Feb 2024 17:25:26 +0100 Subject: [PATCH] Move Explorer tests to scheduled CI (#8305) Signed-off-by: Glenn Jocher --- tests/test_explorer.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/test_explorer.py b/tests/test_explorer.py index 5b11c447..296db949 100644 --- a/tests/test_explorer.py +++ b/tests/test_explorer.py @@ -1,11 +1,13 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license import PIL +import pytest from ultralytics import Explorer from ultralytics.utils import ASSETS +@pytest.mark.slow def test_similarity(): """Test similarity calculations and SQL queries for correctness and response length.""" exp = Explorer() @@ -22,6 +24,7 @@ def test_similarity(): assert len(sql) > 0 +@pytest.mark.slow def test_det(): """Test detection functionalities and ensure the embedding table has bounding boxes.""" exp = Explorer(data="coco8.yaml", model="yolov8n.pt") @@ -34,6 +37,7 @@ def test_det(): assert isinstance(similar, PIL.Image.Image) +@pytest.mark.slow def test_seg(): """Test segmentation functionalities and verify the embedding table includes masks.""" exp = Explorer(data="coco8-seg.yaml", model="yolov8n-seg.pt") @@ -45,6 +49,7 @@ def test_seg(): assert isinstance(similar, PIL.Image.Image) +@pytest.mark.slow def test_pose(): """Test pose estimation functionalities and check the embedding table for keypoints.""" exp = Explorer(data="coco8-pose.yaml", model="yolov8n-pose.pt")