Fix minor error in dynamic load function

scripts/tests/test_validate_docstrings.py

@@ -4,6 +4,8 @@
 import textwrap
 import pytest
 import numpy as np
+import pandas
+
 import validate_docstrings
 validate_one = validate_docstrings.validate_one
 
@@ -1004,6 +1006,32 @@ def test_item_subsection(self, idx, subsection):
         assert result[idx][3] == subsection
 
 
+class TestDocstringClass(object):
+    @pytest.mark.parametrize('name_and_expected_obj',
+                             [('pandas.isnull', pandas.isnull),
+                              ('pandas.DataFrame', pandas.DataFrame),
+                              ('pandas.Series.sum', pandas.Series.sum)])
+    def test_resolves_class_name(self, name_and_expected_obj):
+        name, expected_obj = name_and_expected_obj
+        d = validate_docstrings.Docstring(name)
+        assert d.obj is expected_obj
+
+    @pytest.mark.parametrize('invalid_name', ['panda', 'panda.DataFrame'])
+    def test_raises_for_invalid_module_name(self, invalid_name):
+        # When an invalid module name is supplied, an ImportError should be
+        # raised, and the message should contain the user-supplied name.
+        msg = 'No module can be imported.*{}'.format(invalid_name)
+        with pytest.raises(ImportError, match=msg):
+            validate_docstrings.Docstring(invalid_name)
+
+    @pytest.mark.parametrize('invalid_name',
+                             ['pandas.BadClassName',
+                              'pandas.Series.bad_method_name'])
+    def test_raises_for_invalid_attribute_name(self, invalid_name):
+        with pytest.raises(AttributeError):
+            validate_docstrings.Docstring(invalid_name)
+
+
 class TestMainFunction(object):
     def test_exit_status_for_validate_one(self, monkeypatch):
         monkeypatch.setattr(

scripts/validate_docstrings.py

@@ -267,7 +267,7 @@ def _load_obj(name):
             else:
                 continue
 
-        if 'module' not in locals():
+        if 'obj' not in locals():
             raise ImportError('No module can be imported '
                               'from "{}"'.format(name))