Fix a Python 2 test

2020-03-21 03:13:24 +01:00
parent a2764283ba
commit 28c1ba6c1c
1 changed files with 2 additions and 5 deletions
--- a/test/test_inference/test_gradual/test_typeshed.py
+++ b/test/test_inference/test_gradual/test_typeshed.py
@@ -136,13 +136,10 @@ def test_math(Script):
    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()
    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'


@pytest.mark.parametrize(