Fix a Python 2 test

This commit is contained in:
Dave Halter
2020-03-21 03:13:59 +01:00
parent a2764283ba
commit 28c1ba6c1c
@@ -136,12 +136,9 @@ def test_math(Script):
assert value assert value
def test_type_var(Script, environment): def test_type_var(Script, environment, skip_python2):
def_, = Script('import typing; T = typing.TypeVar("T1")').infer() def_, = Script('import typing; T = typing.TypeVar("T1")').infer()
assert def_.name == 'TypeVar' assert def_.name == 'TypeVar'
if environment.version_info.major == 2:
assert def_.description == 'TypeVar = object()'
else:
assert def_.description == 'class TypeVar' assert def_.description == 'class TypeVar'