add onnx mark to integration tests

microsoft · Dec 19, 2024 · 1af0272 · 1af0272
1 parent f56a996
commit 1af0272
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 4 deletions.
diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -156,7 +156,8 @@ filterwarnings = [
 ]
 timeout = 120
 markers = [
-    "ollama: mark a test as requiring the Ollama service (use \"not ollama\" to skip those tests)"
+    "ollama: mark a test as requiring the Ollama service (use \"not ollama\" to skip those tests)",
+    "onnx: mark a test as requiring the Onnx service (use \"not onnx\" to skip those tests)"
 ]
 
 [tool.ruff]

diff --git a/python/tests/integration/completions/test_chat_completion_with_image_input_text_output.py b/python/tests/integration/completions/test_chat_completion_with_image_input_text_output.py
@@ -118,7 +118,10 @@
                 ChatMessageContent(role=AuthorRole.USER, items=[TextContent(text="Where was it made?")]),
             ],
             {},
-            marks=pytest.mark.skipif(not onnx_setup, reason="Need a Onnx Model setup"),
+            marks=(
+                pytest.mark.skipif(not onnx_setup, reason="Need a Onnx Model setup"),
+                pytest.mark.onnx,
+            ),
             id="onnx_gen_ai_image_input_file",
         ),
         pytest.param(

diff --git a/python/tests/integration/completions/test_chat_completions.py b/python/tests/integration/completions/test_chat_completions.py
@@ -150,7 +150,10 @@ class Reasoning(KernelBaseModel):
                 ChatMessageContent(role=AuthorRole.USER, items=[TextContent(text="How are you today?")]),
             ],
             {},
-            marks=pytest.mark.skipif(not onnx_setup, reason="Need a Onnx Model setup"),
+            marks=(
+                pytest.mark.skipif(not onnx_setup, reason="Need a Onnx Model setup"),
+                pytest.mark.onnx,
+            ),
             id="onnx_gen_ai",
         ),
         # endregion

diff --git a/python/tests/integration/completions/test_text_completion.py b/python/tests/integration/completions/test_text_completion.py
@@ -123,7 +123,10 @@
             {},
             ["<|user|>Repeat the word Hello<|end|><|assistant|>"],
             {},
-            marks=pytest.mark.skipif(not onnx_setup, reason="Need local Onnx setup"),
+            marks=(
+                pytest.mark.skipif(not onnx_setup, reason="Need a Onnx Model setup"),
+                pytest.mark.onnx,
+            ),
             id="onnx_gen_ai_text_completion",
         ),
         pytest.param(