From 45fecabbf1676bf0bec8d9290f12288694667168 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Sun, 23 Mar 2014 13:30:10 +0100 Subject: [PATCH] definition type shouldn't return import, we'd rather want the implementation name -> #340 --- test/test_api/test_api_classes.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/test/test_api/test_api_classes.py b/test/test_api/test_api_classes.py index f0026dca..cd13cf04 100644 --- a/test/test_api/test_api_classes.py +++ b/test/test_api/test_api_classes.py @@ -64,6 +64,20 @@ def test_basedefinition_type(definition): 'generator', 'statement', 'import', 'param') +def test_basedefinition_type_import(): + def get_types(source): + return set([t.type for t in Script(source).completions()]) + + assert get_types('import t') == set(['module']) + assert get_types('import ') == set(['module']) + assert get_types('import datetime; datetime') == set(['module']) + + assert get_types('from datetime import timedelta') == set(['class']) + assert get_types('from datetime import timedelta; timedelta') == set(['class']) + assert get_types('from json import tool') == set(['module']) + assert get_types('from json import tool; tool') == set(['module']) + + def test_function_call_signature_in_doc(): defs = Script(""" def f(x, y=1, z='a'):