From 0aed08b5be0244aef3d8bab7e50376fee1450480 Mon Sep 17 00:00:00 2001 From: Takeshi KOMIYA Date: Sat, 8 Jun 2019 16:09:45 +0900 Subject: Fix #6451: autodoc: generates docs for "optional import"ed modules as variables --- tests/test_pycode_parser.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'tests/test_pycode_parser.py') diff --git a/tests/test_pycode_parser.py b/tests/test_pycode_parser.py index ba9778b80..8f8c6cbab 100644 --- a/tests/test_pycode_parser.py +++ b/tests/test_pycode_parser.py @@ -149,6 +149,21 @@ def test_complex_assignment_py3(): assert parser.definitions == {} +def test_assignment_in_try_clause(): + source = ('try:\n' + ' a = None #: comment\n' + 'except:\n' + ' b = None #: ignored\n' + 'else:\n' + ' c = None #: comment\n') + parser = Parser(source) + parser.parse() + assert parser.comments == {('', 'a'): 'comment', + ('', 'c'): 'comment'} + assert parser.deforders == {'a': 0, + 'c': 1} + + def test_obj_assignment(): source = ('obj = SomeObject() #: some object\n' 'obj.attr = 1 #: attr1\n' -- cgit v1.2.1