Support more complex starred asssignment case

This commit is contained in:
Takeshi KOMIYA
2017-10-19 00:10:46 +09:00
parent 4fdcae0584
commit 1dfe978fcb
2 changed files with 15 additions and 5 deletions

View File

@@ -121,12 +121,17 @@ def test_complex_assignment():
@pytest.mark.skipif(PY2, reason='tests for py3 syntax')
def test_complex_assignment_py3():
source = 'a, *b, c = (1, 2, 3, 4) #: unpack assignment\n'
source = ('a, *b, c = (1, 2, 3, 4) #: unpack assignment\n'
'd, *self.attr = (5, 6, 7) #: unpack assignment2\n'
'e, *f[0] = (8, 9, 0) #: unpack assignment3\n'
)
parser = Parser(source)
parser.parse()
assert parser.comments == {('', 'a'): 'unpack assignment',
('', 'b'): 'unpack assignment',
('', 'c'): 'unpack assignment',
('', 'd'): 'unpack assignment2',
('', 'e'): 'unpack assignment3',
}
assert parser.definitions == {}