fix bpython#583

thomasballinger · thomasballinger · commit 460d7f0e3a68 · 2015-12-07T00:04:13.000-05:00
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -579,7 +579,7 @@ def set_docstring(self):
         self.docstring = None
         if not self.get_args():
             self.funcprops = None
-        elif self.current_func is not None:
+        if self.current_func is not None:
             try:
                 self.docstring = pydoc.getdoc(self.current_func)
             except IndexError:
diff --git a/bpython/test/test_repl.py b/bpython/test/test_repl.py
@@ -222,6 +222,13 @@ def test_issue572(self):
         self.set_input_line("WonderfulSpam(")
         self.assertTrue(self.repl.get_args())
 
+    def test_issue583(self):
+        self.repl = FakeRepl()
+        self.repl.push("a = 1.2\n", False)
+        self.set_input_line("a.is_integer(")
+        self.repl.set_docstring()
+        self.assertIsNot(self.repl.docstring, None)
+
 
 class TestGetSource(unittest.TestCase):
     def setUp(self):
