Merge pull request #4157 from tk0miya/4156_parse_class_comment

Fix #4156: failed to parse class comment

1 files changed, 12 insertions, 0 deletions

diff --git a/tests/test_pycode_parser.py b/tests/test_pycode_parser.py
index d794376fa..b9327999b 100644
--- a/tests/test_pycode_parser.py
+++ b/tests/test_pycode_parser.py
@@ -246,6 +246,18 @@ def test_nested_class():
                                 'Foo.Bar.attr2': 3}
 
 
+def test_class_comment():
+    source = ('import logging\n'
+              'logger = logging.getLogger(__name__)\n'
+              '\n'
+              'class Foo(object):\n'
+              '    """Bar"""\n')
+    parser = Parser(source)
+    parser.parse()
+    assert parser.comments == {}
+    assert parser.definitions == {'Foo': ('class', 4, 5)}
+
+
 def test_comment_picker_multiline_string():
     source = ('class Foo(object):\n'
               '    a = None\n'