Handle missing optional pyvistaqt package

bjlittle · Oct 16, 2023 · 1e621c6 · 1e621c6
1 parent bcdb488
commit 1e621c6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/geovista/qt.py b/src/geovista/qt.py
@@ -11,7 +11,7 @@
 
 try:
     import pyvistaqt as pvqt
-except ImportError as e:
+except ModuleNotFoundError as e:
     e.msg = f'{e.msg} - please install the "pyvistaqt" and "pyqt" packages.'
     raise
 

diff --git a/tests/test_qt.py b/tests/test_qt.py
@@ -15,12 +15,12 @@ def test_pyvistaqt_import():
     """
     try:
         import pyvistaqt
-    except ImportError:
+    except ModuleNotFoundError:
         pyvistaqt = False
 
     if not pyvistaqt:
         emsg = 'please install the "pyvistaqt" and "pyqt" packages'
-        with pytest.raises(ImportError, match=emsg):
+        with pytest.raises(ModuleNotFoundError, match=emsg):
             import geovista.qt  # noqa: F401