Provide message if test data is not installed

mdboom · mdboom · commit 2469db563fe7 · 2015-10-19T09:57:44.000-04:00
diff --git a/lib/matplotlib/__init__.py b/lib/matplotlib/__init__.py
@@ -1484,6 +1484,9 @@ def tk_window_focus():
 
 
 def verify_test_dependencies():
+    if not os.path.isdir(os.path.join(os.path.dirname(__file__), 'tests')):
+        raise ImportError("matplotlib test data is not installed")
+
     try:
         import nose
         try:
diff --git a/tests.py b/tests.py
@@ -45,6 +45,10 @@ def run(extra_args):
     else:
         faulthandler.enable()
 
+    if not os.path.isdir(
+            os.path.join(os.path.dirname(matplotlib.__file__), 'tests')):
+        raise ImportError("matplotlib test data is not installed")
+
     nose.main(addplugins=[x() for x in plugins],
               defaultTest=default_test_modules,
               argv=sys.argv + extra_args)
