Skip yaml-dependent tests if yaml isn't installed

SINTEF · Aug 18, 2023 · 0e01166 · 0e01166
1 parent ce07ca0
commit 0e01166
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/bindings/python/tests/test_storage.py b/bindings/python/tests/test_storage.py
@@ -10,6 +10,12 @@
 except ModuleNotFoundError:
     HAVE_PYTEST = False
 
+try:
+    import yaml
+    HAVE_YAML = True
+except ModuleNotFoundError:
+    HAVE_YAML = False
+
 
 thisdir = os.path.abspath(os.path.dirname(__file__))
 
@@ -45,11 +51,7 @@
 
 
 # Test yaml
-try:
-    import yaml
-except ImportError:
-    pass
-else:
+if HAVE_YAML:
     print('--- testing yaml')
     inst.save('yaml://inst.yaml?mode=w')
     del inst
@@ -104,7 +106,8 @@
 
 
 # Tests for issue #587
-bytearr = inst.to_bytes("yaml")
+if HAVE_YAML:
+    bytearr = inst.to_bytes("yaml")
 #print(bytes(bytearr).decode())
 if HAVE_PYTEST:
     with pytest.raises(dlite.DLiteError):