From eeeeee49f52ddfd1ab6a08c10bcf54c94a4d64fc Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Fri, 21 Mar 2014 11:12:40 +0100 Subject: [PATCH] new tests for that whole import mess --- test/test_evaluate/test_imports.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/test/test_evaluate/test_imports.py b/test/test_evaluate/test_imports.py index 17f268df..4954db59 100644 --- a/test/test_evaluate/test_imports.py +++ b/test/test_evaluate/test_imports.py @@ -41,3 +41,20 @@ def test_follow_import_incomplete(): datetime = jedi.Script("import datetim").completions()[0] definition = datetime.follow_definition()[0] assert definition + + # empty `from * import` parts + datetime = jedi.Script("from datetime import ").completions()[0] + definitions = datetime.follow_definition() + assert [d.type for d in definitions if d.name == 'date'] == ['class'] + + # incomplete `from * import` part + datetime = jedi.Script("from datetime import datetim").completions()[0] + definition = datetime.follow_definition() + assert [d.type for d in definitions] == ['class'] + + +def test_follow_definition_land_on_import(): + datetime = jedi.Script("import datetime; datetim").completions()[0] + definition = datetime.follow_definition()[0] + print(datetime._definition, definition) + assert definition.type == 'module'