Skip tests with missing requirements

SINTEF · Dec 2, 2024 · 25d0092 · 25d0092
1 parent 668de12
commit 25d0092
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 11 deletions.
diff --git a/examples/read-csv/main.py b/examples/read-csv/main.py
@@ -1,8 +1 @@
-try:
-    import pandas  # noqa: F401
-    import tables  # noqa: F401
-except ImportError:
-    import sys
-    sys.exit(44)  # skip this test if pandas is not available
-
 import readcsv  # noqa: F401
diff --git a/examples/read-csv/readcsv.py b/examples/read-csv/readcsv.py
@@ -1,6 +1,11 @@
 from pathlib import Path
 
 import dlite
+from dlite import importskip
+
+importskip("pandas")
+importskip("tables")
+importskip("yaml")
 
 
 # Set up some paths

diff --git a/storages/python/tests-python/test_image.py b/storages/python/tests-python/test_image.py
@@ -2,12 +2,12 @@
 from pathlib import Path
 
 import numpy as np
+
 import dlite
+from dlite.testutils import importskip
 
-try:
-    import skimage
-except ImportError:
-    sys.exit(44)  # skip test
+importskip("scipy")
+importskip("skimage")
 
 
 thisdir = Path(__file__).absolute().parent