One more small test change

This commit is contained in:
Dave Halter
2019-05-06 23:59:39 +02:00
parent ea6462daf4
commit a6693616a0

View File

@@ -34,10 +34,7 @@ def test_follow_import_incomplete(Script, environment):
# incomplete `from * import` part
datetime = check_follow_definition_types(Script, "from datetime import datetim")
if environment.version_info.major == 2:
assert datetime == ['class']
else:
assert set(datetime) == {'class', 'instance'} # py3: builtin and pure py version
assert datetime == ['class']
# os.path check
ospath = check_follow_definition_types(Script, "from os.path import abspat")
assert ospath == ['function']