add slow marker to skip skip slow tests locally

rapidsai · Nov 7, 2023 · f9c651b · f9c651b
1 parent a59b904
commit f9c651b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,7 +1,8 @@
 import pytest
 
 
-def pytest_collection_modifyitems(items):
+def pytest_collection_modifyitems(config, items):
+    skip_slow = pytest.mark.skip(reason="need --runslow option to run")
     for item in items:
         path = item.location[0]
         if "/tf_backend/" in path:
@@ -14,3 +15,5 @@ def pytest_collection_modifyitems(items):
             item.add_marker(pytest.mark.tensorflow)
             item.add_marker(pytest.mark.pytorch)
             item.add_marker(pytest.mark.jax)
+        if "slow" in item.keywords:
+            item.add_marker(skip_slow)
diff --git a/tests/report/beir/test_report.py b/tests/report/beir/test_report.py
@@ -82,6 +82,7 @@ def test_no_invalid_scores(
     assert not np.isinf(ndcg).any()
 
 
+@pytest.mark.slow
 @pytest.mark.benchmark(
     warmup_iterations=1,
 )