TST: Fix ImportError: No module named 'mpl_toolkits'

cgohlke · cgohlke · commit 82895142344c · 2014-07-17T21:21:18.000-07:00
Apparently `imp.find_module` does not support namespace packages. Fixes issue matplotlib#3252.
diff --git a/lib/matplotlib/testing/decorators.py b/lib/matplotlib/testing/decorators.py
@@ -308,10 +308,15 @@ def find_dotted_module(module_name, path=None):
             """A version of imp which can handle dots in the module name"""
             res = None
             for sub_mod in module_name.split('.'):
-                res = file, path, _ = imp.find_module(sub_mod, path)
-                path = [path]
-                if file is not None:
-                    file.close()
+                try:
+                    res = file, path, _ = imp.find_module(sub_mod, path)
+                    path = [path]
+                    if file is not None:
+                        file.close()
+                except ImportError:
+                    # assume namespace package
+                    path = sys.modules[sub_mod].__path__
+                    res = None, path, None
             return res
 
         mod_file = find_dotted_module(func.__module__)[1]
