skip tests that require pandas if pandas not available

DistrictDataLabs · Jan 15, 2021 · 98d1920 · 98d1920
1 parent 06206db
commit 98d1920
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/test_contrib/test_missing/test_bar.py b/tests/test_contrib/test_missing/test_bar.py
@@ -47,6 +47,7 @@ class TestMissingBarVisualizer(VisualTestCase):
     FeatureImportances visualizer
     """
 
+    @pytest.mark.skipif(pd is None, reason="pandas is required")
     def test_missingvaluesbar_pandas(self):
         """
         Integration test of visualizer with pandas

diff --git a/tests/test_contrib/test_missing/test_dispersion.py b/tests/test_contrib/test_missing/test_dispersion.py
@@ -47,6 +47,7 @@ class TestMissingValuesDispersion(VisualTestCase):
     MissingValuesDispersion visualizer
     """
 
+    @pytest.mark.skipif(pd is None, reason="pandas is required")
     def test_missingvaluesdispersion_with_pandas(self):
         """
         Integration test of visualizer with pandas
@@ -72,6 +73,7 @@ def test_missingvaluesdispersion_with_pandas(self):
 
         self.assert_images_similar(viz, tol=self.tol)
 
+    @pytest.mark.skipif(pd is None, reason="pandas is required")
     def test_missingvaluesdispersion_with_pandas_with_y_targets(self):
         """
         Integration test of visualizer with pandas with y targets