Skip test also if bloch_sphere_visualization extra installed

qutech · Jul 3, 2020 · 8a6bef1 · 8a6bef1
1 parent ddc76bf
commit 8a6bef1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_extras.py b/tests/test_extras.py
@@ -43,7 +43,8 @@ def test_fancy_progressbar_not_available(self):
         self.assertIs(tqdm, util._tqdm)
 
     @pytest.mark.skipif(
-        'plotting' in os.environ.get('INSTALL_EXTRAS', all_extras),
+        any(extra in os.environ.get('INSTALL_EXTRAS', all_extras)
+            for extra in ['plotting', 'bloch_sphere_visualization']),
         reason='Skipping tests for missing plotting extra in build with matplotlib')  # noqa
     def test_plotting_not_available(self):
         with self.assertRaises(ModuleNotFoundError):