From 8e3f85c4752c25dd032459a40fe00a5e90f5d51d Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Fri, 17 May 2019 23:49:26 +0200 Subject: [PATCH] Revert "One more small test change" This reverts commit a6693616a094e1cadb5c35d145581e805f0a89da. --- test/test_api/test_api_classes_follow_definition.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/test_api/test_api_classes_follow_definition.py b/test/test_api/test_api_classes_follow_definition.py index d022c69c..1b6cf793 100644 --- a/test/test_api/test_api_classes_follow_definition.py +++ b/test/test_api/test_api_classes_follow_definition.py @@ -34,7 +34,10 @@ def test_follow_import_incomplete(Script, environment): # incomplete `from * import` part datetime = check_follow_definition_types(Script, "from datetime import datetim") - assert datetime == ['class'] + if environment.version_info.major == 2: + assert datetime == ['class'] + else: + assert set(datetime) == {'class', 'instance'} # py3: builtin and pure py version # os.path check ospath = check_follow_definition_types(Script, "from os.path import abspat") assert ospath == ['function']