test_ndarray.py - skip if numpy is not installed

idanmiara · Nov 29, 2022 · e3dc5dc · e3dc5dc
1 parent 7ea6cf2
commit e3dc5dc
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/core/test_ndarray.py b/tests/core/test_ndarray.py
@@ -1,9 +1,12 @@
 from dataclasses import dataclass
 from typing import Sequence, TypeVar
 
-import numpy
-import numpy.typing
-import numpy.testing
+try:
+    import numpy
+    import numpy.typing
+    import numpy.testing
+except ImportError:
+    pytest.skip("numpy not available", allow_module_level=True)
 
 from dcd import from_dict, Config
 from tests.common import ndarray_support