Fix deprecations in tests

2020-03-21 02:42:00 +01:00
parent 5f6a25fb58
commit 5b54ac835d
5 changed files with 9 additions and 5 deletions
--- a/test/test_inference/test_gradual/test_typeshed.py
+++ b/test/test_inference/test_gradual/test_typeshed.py
@@ -136,10 +136,13 @@ def test_math(Script):
    assert value


-def test_type_var(Script):
+def test_type_var(Script, environment):
    def_, = Script('import typing; T = typing.TypeVar("T1")').infer()
    assert def_.name == 'TypeVar'
-    assert def_.description == 'class TypeVar'
+    if environment.version_info.major == 2:
+        assert def_.description == 'TypeVar = object()'
+    else:
+        assert def_.description == 'class TypeVar'


@pytest.mark.parametrize(