Catch exception during test discovery

qutech · Jul 3, 2020 · b65d79d · b65d79d
1 parent 1bb09ae
commit b65d79d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_extras.py b/tests/test_extras.py
@@ -55,7 +55,10 @@ def test_plotting_not_available(self):
         reason='Skipping tests for missing bloch sphere visualization tests in build with qutip')  # noqa
     def test_bloch_sphere_visualization_not_available(self):
 
-        from filter_functions import plotting
+        try:
+            from filter_functions import plotting
+        except ModuleNotFoundError:
+            plotting = None
 
         with self.assertRaises(RuntimeError):
             plotting.get_bloch_vector(testutil.rng.standard_normal((10, 2)))