Implement support for TypeVar inference for __new__

jedi/inference/value/klass.py

@@ -376,19 +376,8 @@ class ClassMixin:
             if sigs:
                 return sigs
         args = ValuesArguments([])
-        init_funcs = self.py__call__(args).py__getattribute__('__init__')
-        if len(init_funcs) == 1:
-            init = next(iter(init_funcs))
-            try:
-                class_context = init.class_context
-            except AttributeError:
-                pass
-            else:
-                # In the case where we are on object.__init__, we try to use
-                # __new__.
-                if class_context.get_root_context().is_builtins_module() \
-                        and init.class_context.name.string_name == "object":
-                    init_funcs = self.py__call__(args).py__getattribute__('__new__')
+        instance = self.py__call__(args)
+        init_funcs = init_or_new_func(instance)
 
         dataclass_sigs = self._get_dataclass_transform_signatures()
         if dataclass_sigs:
@@ -481,6 +470,23 @@ class ClassMixin:
         return ValueSet({self})
 
 
+def init_or_new_func(value):
+    init_funcs = value.py__getattribute__('__init__')
+    if len(init_funcs) == 1:
+        init = next(iter(init_funcs))
+        try:
+            class_context = init.class_context
+        except AttributeError:
+            pass
+        else:
+            # In the case where we are on object.__init__, we try to use
+            # __new__.
+            if class_context.get_root_context().is_builtins_module() \
+                    and init.class_context.name.string_name == "object":
+                return value.py__getattribute__('__new__')
+    return init_funcs
+
+
 class DataclassParamName(BaseTreeParamName):
     """
     Represent a field declaration on a class with dataclass semantics.