Updated Manifold to extend Projection Visualizer (#930)

DistrictDataLabs · Jul 27, 2019 · c6ee8ee · c6ee8ee
1 parent b10fc17
commit c6ee8ee
Show file tree

Hide file tree

Showing 12 changed files with 483 additions and 343 deletions.
diff --git a/tests/baseline_images/test_features/test_manifold/test_manifold_classification.png b/tests/baseline_images/test_features/test_manifold/test_manifold_classification.png
diff --git a/...baseline_images/test_features/test_manifold/test_manifold_classification_3d.png b/...baseline_images/test_features/test_manifold/test_manifold_classification_3d.png
diff --git a/tests/baseline_images/test_features/test_manifold/test_manifold_pandas.png b/tests/baseline_images/test_features/test_manifold/test_manifold_pandas.png
diff --git a/tests/baseline_images/test_features/test_manifold/test_manifold_regression.png b/tests/baseline_images/test_features/test_manifold/test_manifold_regression.png
diff --git a/tests/baseline_images/test_features/test_manifold/test_manifold_regression_3d.png b/tests/baseline_images/test_features/test_manifold/test_manifold_regression_3d.png
diff --git a/tests/baseline_images/test_features/test_manifold/test_manifold_single.png b/tests/baseline_images/test_features/test_manifold/test_manifold_single.png
diff --git a/tests/baseline_images/test_features/test_manifold/test_manifold_single_3d.png b/tests/baseline_images/test_features/test_manifold/test_manifold_single_3d.png
diff --git a/tests/test_exceptions.py b/tests/test_exceptions.py
@@ -0,0 +1,18 @@
+##########################################################################
+## Imports
+##########################################################################
+import pytest
+from yellowbrick.exceptions import NotFitted
+
+##########################################################################
+## NotFitted Exception Tests
+##########################################################################
+class TestNotFitted(object):
+    """
+    Test Not Fitted Error
+    """    
+    def test_correct_message(self):
+        "Ensure that NotFitted error is raised properly"
+        msg = "instance is not fitted yet, please call fit"
+        with pytest.raises(NotFitted, match=msg):
+            raise NotFitted.from_estimator(self, 'transform')