Some more tests

Author: ilevkivskyi

test-data/unit/check-classes.test

@@ -7677,10 +7677,13 @@ class A:
 
 T = TypeVar("T")
 def foo(f: Callable[[], T]) -> T: ...
+def bar(t: type[T]) -> T: ...
 
 ta: type[A]
 reveal_type(foo(A))  # N: Revealed type is "builtins.int"
 reveal_type(foo(ta))  # N: Revealed type is "builtins.int"
+reveal_type(bar(A))  # N: Revealed type is "__main__.A"
+reveal_type(bar(ta))  # N: Revealed type is "__main__.A"
 
 [case testNewReturnTypeCallableNarrowing]
 from typing import Callable, TypeVar
@@ -7700,6 +7703,40 @@ if issubclass(tx, B):
     reveal_type(tx)  # N: Revealed type is "type[__main__.B]"
 [builtins fixtures/isinstancelist.pyi]
 
+[case testNewReturnTypeOverloadNarrowing]
+from typing import TypeVar, Generic, Union, overload
+
+T = TypeVar("T", int, str)
+
+class Series(Generic[T]):
+    @overload
+    def __new__(cls, dtype: int) -> IntSeries: ...
+    @overload
+    def __new__(cls, dtype: type[T]) -> Series[T]: ...
+    @overload
+    def __new__(cls, dtype=...) -> Series: ...
+    def __new__(cls, dtype=...) -> Series:
+        ...
+
+class IntSeries(Series[int]): ...
+
+class Index:
+    @overload
+    def __new__(cls, dtype: int) -> IntIndex: ...
+    @overload
+    def __new__(cls, dtype=...) -> Index: ...
+    def __new__(cls, dtype=...) -> Index:
+        ...
+
+class IntIndex(Index): ...
+
+def foo(x: Union[Series, int, Index]):
+    if isinstance(x, (Series, Index)):
+        reveal_type(x)  # N: Revealed type is "__main__.Series[Any] | __main__.Index"
+    else:
+        reveal_type(x)  # N: Revealed type is "builtins.int"
+[builtins fixtures/isinstancelist.pyi]
+
 [case testMetaclassPlaceholderNode]
 from sympy.assumptions import ManagedProperties
 from sympy.ops import AssocOp