resolving precommit checks

opencomputeproject · Sep 20, 2023 · 92986e0 · 92986e0
1 parent 2a7c0df
commit 92986e0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/api/test_udf.py b/tests/api/test_udf.py
@@ -1,6 +1,14 @@
 from pprint import pprint
+import pytest
 
 
+@pytest.fixture(scope='module', autouse=True)
+def skip_all(testbed_instance):
+    testbed = testbed_instance
+    if testbed is not None and len(testbed.npu) != 1:
+        pytest.skip('invalid for {} testbed'.format(testbed.name))
+
+@pytest.mark.npu
 class TestSaiUdf:
     # object with parent SAI_OBJECT_TYPE_UDF_MATCH SAI_OBJECT_TYPE_UDF_GROUP