From 0d9f9f0e4ae9f7ad1822c950ba99618823d13965 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Mon, 7 Apr 2014 14:54:56 +0200 Subject: [PATCH] parent on completions tests --- test/test_api/test_api_classes.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/test/test_api/test_api_classes.py b/test/test_api/test_api_classes.py index d0b2f9c2..3f90db21 100644 --- a/test/test_api/test_api_classes.py +++ b/test/test_api/test_api_classes.py @@ -168,3 +168,15 @@ class TestParent(TestCase): pass''', 1, len('def spam')) assert parent.name == 'spam' assert parent.parent().type == 'module' + + def test_parent_on_completion(self): + parent = Script(dedent('''\ + class Foo(): + def bar(): pass + Foo().bar''')).completions()[0].parent() + assert parent.name == 'Foo' + assert parent.type == 'class' + + parent = Script('str.join').completions()[0].parent() + assert parent.name == 'str' + assert parent.type == 'class'