adding a pytest mark "cufile" for tests that requires cufile

rapidsai · Oct 27, 2024 · 77826ee · 77826ee
1 parent 032c326
commit 77826ee
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/python/kvikio/pyproject.toml b/python/kvikio/pyproject.toml
@@ -144,3 +144,6 @@ filterwarnings = [
     "ignore:Jitify is performing a one-time only warm-up to populate the persistent cache",
     "ignore::DeprecationWarning:botocore.*",
 ]
+markers = [
+    "cufile: tests to skip if cuFile isn't available e.g. run with `pytest -m 'not cufile'`"
+]
diff --git a/python/kvikio/tests/test_cufile_driver.py b/python/kvikio/tests/test_cufile_driver.py
@@ -1,10 +1,12 @@
 # Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
 # See file LICENSE for terms.
 
+import pytest
+
 import kvikio.cufile_driver
-import kvikio.defaults
 
 
+@pytest.mark.cufile
 def test_open_and_close():
     kvikio.cufile_driver.driver_open()
     kvikio.cufile_driver.driver_close()