fix: ClassTypeHint.recurse_bases() now parameterizes type parameters.

Author: NiklasRosenstein

.changelog/_unreleased.toml

@@ -0,0 +1,5 @@
+[[entries]]
+id = "984c08a5-3fa1-40d7-baa7-3886f8fd2548"
+type = "fix"
+description = "`ClassTypeHint.recurse_bases()` now parameterizes type parameters."
+author = "@NiklasRosenstein"

src/typeapi/typehint.py

@@ -320,7 +320,10 @@ def recurse_bases(
             if response == "skip":
                 continue
 
-            current_bases = cast(List[ClassTypeHint], [TypeHint(x, current.type).evaluate() for x in current.bases])
+            current_bases = cast(
+                List[ClassTypeHint],
+                [TypeHint(x, current.type).evaluate().parameterize(current.get_parameter_map()) for x in current.bases],
+            )
 
             if order == "bfs":
                 bases.extend(current_bases)

src/typeapi/typehint_test.py

@@ -502,7 +502,7 @@ def test__TypeHint__repeated() -> None:
     assert hint.repeated
 
 
-def test__ClassTypeHint__iter_all_bases() -> None:
+def test__ClassTypeHint__recurse_bases_generic() -> None:
     class A:
         pass
 
@@ -512,6 +512,9 @@ class B(A, int):
     class C(B, Generic[T]):
         pass
 
+    class D(C[T]):
+        pass
+
     hint = TypeHint(C)
     assert isinstance(hint, ClassTypeHint)
     assert hint.type is C
@@ -536,3 +539,44 @@ class C(B, Generic[T]):
         TypeHint(Generic[T]),
         TypeHint(object),
     ]
+
+    hint = TypeHint(D[int])
+    assert isinstance(hint, ClassTypeHint)
+    assert hint.type is D
+    assert hint.bases == (C[T],)
+    assert list(hint.recurse_bases("bfs")) == [
+        TypeHint(D[int]),
+        TypeHint(C[int]),
+        TypeHint(B),
+        TypeHint(Generic[T]),
+        TypeHint(A),
+        TypeHint(int),
+        TypeHint(object),
+        TypeHint(object),
+        TypeHint(object),
+    ]
+
+
+def test__ClassTypeHint__recurse_bases_for_list_subclass() -> None:
+    class CustomList(List[T]):
+        pass
+
+    hint = TypeHint(CustomList)
+    assert isinstance(hint, ClassTypeHint)
+    assert hint.type is CustomList
+    assert hint.bases == (List[T],)
+    assert list(hint.recurse_bases("bfs")) == [
+        TypeHint(CustomList),
+        TypeHint(List[T]),
+        TypeHint(object),
+    ]
+
+    hint = TypeHint(CustomList[int])
+    assert isinstance(hint, ClassTypeHint)
+    assert hint.type is CustomList
+    assert hint.bases == (List[T],)
+    assert list(hint.recurse_bases("bfs")) == [
+        TypeHint(CustomList[int]),
+        TypeHint(List[int]),
+        TypeHint(object),
+    ]