From 1c1b51f116f334395836cb7adb8ce2d02d143f7a Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Mon, 24 Mar 2014 01:31:14 +0100 Subject: [PATCH] more type tests --- test/test_api/test_api_classes.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/test/test_api/test_api_classes.py b/test/test_api/test_api_classes.py index cd13cf04..82c65ebb 100644 --- a/test/test_api/test_api_classes.py +++ b/test/test_api/test_api_classes.py @@ -65,18 +65,27 @@ def test_basedefinition_type(definition): def test_basedefinition_type_import(): - def get_types(source): - return set([t.type for t in Script(source).completions()]) + def get_types(source, **kwargs): + return set([t.type for t in Script(source, **kwargs).completions()]) + # import one level assert get_types('import t') == set(['module']) assert get_types('import ') == set(['module']) assert get_types('import datetime; datetime') == set(['module']) + # from 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']) + # import two levels + assert get_types('import json.tool; json') == set(['module']) + assert get_types('import json.tool; json.tool') == set(['module']) + assert get_types('import json.tool; json.tool.main') == set(['function']) + assert get_types('import json.tool') == set(['module']) + assert get_types('import json.tool', column=9) == set(['module']) + def test_function_call_signature_in_doc(): defs = Script("""