diff --git a/test/test_api/test_api_classes_follow_definition.py b/test/test_api/test_api_classes_follow_definition.py index 0d74338b..ffc0cb64 100644 --- a/test/test_api/test_api_classes_follow_definition.py +++ b/test/test_api/test_api_classes_follow_definition.py @@ -34,7 +34,7 @@ def test_follow_import_incomplete(): # incomplete `from * import` part datetime = check_follow_definition_types("from datetime import datetim") - assert set(datetime) == set(['class']) # py33: builtin and pure py version + assert set(datetime) == set(['class', 'instance']) # py33: builtin and pure py version # os.path check ospath = check_follow_definition_types("from os.path import abspat") @@ -54,7 +54,7 @@ def test_follow_definition_nested_import(): assert types == ['module'] types = check_follow_definition_types("import pkg.mod1; pkg.mod1.a") - assert types == ['class'] + assert types == ['instance'] def test_follow_definition_land_on_import(): diff --git a/test/test_evaluate/test_compiled.py b/test/test_evaluate/test_compiled.py index e238cd81..0852a5e1 100644 --- a/test/test_evaluate/test_compiled.py +++ b/test/test_evaluate/test_compiled.py @@ -1,6 +1,6 @@ from jedi._compatibility import builtins, is_py3 from jedi.parser.representation import Function -from jedi.evaluate import compiled +from jedi.evaluate import compiled, representation from jedi.evaluate import Evaluator from jedi import Script @@ -13,7 +13,7 @@ def test_simple(): assert len(upper) == 1 objs = list(e.execute(upper[0])) assert len(objs) == 1 - assert objs[0].obj is str + assert isinstance(objs[0], representation.Instance) def test_fake_loading():