Adds testcases for psf#4254

sumezulike · Feb 27, 2024 · cbb6316 · cbb6316
1 parent 8990023
commit cbb6316
Showing 1 changed file with 119 additions and 0 deletions.
diff --git a/tests/data/cases/type_params.py b/tests/data/cases/type_params.py
@@ -13,6 +13,33 @@ def it_gets_worse[WhatIsTheLongestTypeVarNameYouCanThinkOfEnoughToMakeBlackSplit
 
 def magic[Trailing, Comma,](): pass
 
+# Combinations of type params, function params and return types
+def func00[T](a): pass
+
+def func01[T](a,): pass
+
+def func10[T,](a): pass
+
+def func11[T,](a,): pass
+
+def func000[T](a) -> list[A, B]: pass
+
+def func010[T](a,) -> list[A, B]: pass
+
+def func100[T,](a) -> list[A, B]: pass
+
+def func110[T,](a,) -> list[A, B]: pass
+
+def func001[T](a) -> list[A, B,]: pass
+
+def func011[T](a,) -> list[A, B,]: pass
+
+def func101[T,](a) -> list[A, B,]: pass
+
+def func111[T,](a,) -> list[A, B,]: pass
+
+def long_func_params[T](aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa, bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb): pass
+
 # output
 
 
@@ -56,3 +83,95 @@ def magic[
     Comma,
 ]():
     pass
+
+
+# Combinations of type params, function params and return types
+def func00[T](a):
+    pass
+
+
+def func01[T](
+    a,
+):
+    pass
+
+
+def func10[
+    T,
+](a):
+    pass
+
+
+def func11[
+    T,
+](
+    a,
+):
+    pass
+
+
+def func000[T](a) -> list[A, B]:
+    pass
+
+
+def func010[T](
+    a,
+) -> list[A, B]:
+    pass
+
+
+def func100[
+    T,
+](a) -> list[A, B]:
+    pass
+
+
+def func110[
+    T,
+](
+    a,
+) -> list[A, B]:
+    pass
+
+
+def func001[T](a) -> list[
+    A,
+    B,
+]:
+    pass
+
+
+def func011[T](
+    a,
+) -> list[
+    A,
+    B,
+]:
+    pass
+
+
+def func101[
+    T,
+](a) -> list[
+    A,
+    B,
+]:
+    pass
+
+
+def func111[
+    T,
+](
+    a,
+) -> list[
+    A,
+    B,
+]:
+    pass
+
+
+def long_func_params[T](
+    aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa,
+    bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb,
+):
+    pass