From 0202d4ed0ad167c294f8306e120a219d8effc4f4 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Sun, 22 Dec 2019 14:32:07 +0100 Subject: [PATCH] Test parents a bit better --- test/test_api/test_classes.py | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/test/test_api/test_classes.py b/test/test_api/test_classes.py index 7251426e..9e4ca5aa 100644 --- a/test/test_api/test_classes.py +++ b/test/test_api/test_classes.py @@ -277,10 +277,24 @@ def test_parent_on_function(Script): def test_parent_on_completion(Script): - parent = Script(dedent('''\ + script = Script(dedent('''\ class Foo(): - def bar(): pass - Foo().bar''')).complete()[0].parent() + def bar(name): name + Foo().bar''')) + + bar, = script.complete() + parent = bar.parent() + assert parent.name == 'Foo' + assert parent.type == 'class' + + param, = script.goto(line=2) + parent = param.parent() + assert parent.name == 'Foo' + assert parent.type == 'class' + + name, = [d for d in script.names(all_scopes=True, references=True) + if d.name == 'name' and d.type == 'statement'] + parent = param.parent() assert parent.name == 'Foo' assert parent.type == 'class'