Fix some stuff to make parso work again.

This commit is contained in:
Dave Halter
2017-06-02 00:00:31 +02:00
parent af26cc9f05
commit e0485b032e
6 changed files with 11 additions and 9 deletions

View File

@@ -11,8 +11,10 @@ import pytest
class TestCallAndName():
def get_call(self, source):
# Get the simple_stmt and then the first one.
simple_stmt = parse(source).children[0]
return simple_stmt.children[0]
node = parse(source).children[0]
if node.type == 'simple_stmt':
return node.children[0]
return node
def test_name_and_call_positions(self):
name = self.get_call('name\nsomething_else')