DistrictDataLabs · bbengfort · Jul 27, 2019 · Jul 22, 2019 · Jul 23, 2019 · Jul 23, 2019
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')