refactor tests

Author: KevinRK29

mypy/test/test_nativeparse.py

@@ -52,7 +52,11 @@
 class NativeParserSuite(DataSuite):
     required_out_section = True
     base_path = "."
-    files = ["native-parser.test"] if has_nativeparse else []
+    files = (
+        ["native-parser.test", "native-parser-python311.test", "native-parser-python312.test"]
+        if has_nativeparse
+        else []
+    )
 
     def run_case(self, testcase: DataDrivenTestCase) -> None:
         test_parser(testcase)
@@ -77,6 +81,8 @@ def test_parser(testcase: DataDrivenTestCase) -> None:
 
     if testcase.file.endswith("python310.test"):
         options.python_version = (3, 10)
+    elif testcase.file.endswith("python311.test"):
+        options.python_version = (3, 11)
     elif testcase.file.endswith("python312.test"):
         options.python_version = (3, 12)
     elif testcase.file.endswith("python313.test"):

test-data/unit/native-parser-python311.test

@@ -0,0 +1,110 @@
+-- Native parser test cases that target Python 3.11.
+
+[case testTryExceptStar]
+try:
+    x
+except* ValueError:
+    y
+[out]
+MypyFile:1(
+  TryStmt:1(
+    Block:2(
+      ExpressionStmt:2(
+        NameExpr(x)))
+    *
+    NameExpr(ValueError)
+    Block:4(
+      ExpressionStmt:4(
+        NameExpr(y)))))
+
+[case testTryExceptStarWithVar]
+try:
+    x
+except* Exception as e:
+    y
+[out]
+MypyFile:1(
+  TryStmt:1(
+    Block:2(
+      ExpressionStmt:2(
+        NameExpr(x)))
+    *
+    NameExpr(Exception)
+    NameExpr(e)
+    Block:4(
+      ExpressionStmt:4(
+        NameExpr(y)))))
+
+[case testTryMultipleExceptStar]
+try:
+    x
+except* ValueError:
+    y
+except* KeyError as e:
+    z
+[out]
+MypyFile:1(
+  TryStmt:1(
+    Block:2(
+      ExpressionStmt:2(
+        NameExpr(x)))
+    *
+    NameExpr(ValueError)
+    Block:4(
+      ExpressionStmt:4(
+        NameExpr(y)))
+    NameExpr(KeyError)
+    NameExpr(e)
+    Block:6(
+      ExpressionStmt:6(
+        NameExpr(z)))))
+
+[case testTryExceptStarElseFinally]
+try:
+    x
+except* ValueError:
+    y
+else:
+    z
+finally:
+    w
+[out]
+MypyFile:1(
+  TryStmt:1(
+    Block:2(
+      ExpressionStmt:2(
+        NameExpr(x)))
+    *
+    NameExpr(ValueError)
+    Block:4(
+      ExpressionStmt:4(
+        NameExpr(y)))
+    Else(
+      ExpressionStmt:6(
+        NameExpr(z)))
+    Finally(
+      ExpressionStmt:8(
+        NameExpr(w)))))
+
+[case testUnpackTypeInSignature]
+def f(*args: *Ts) -> None:
+    pass
+[out]
+MypyFile:1(
+  FuncDef:1(
+    f
+    def (*args: *Ts?) -> None?
+    VarArg(
+      Var(args))
+    Block:2(
+      PassStmt:2())))
+
+[case testUnpackTypeInTuple]
+x: tuple[int, *Ts, str]
+[out]
+MypyFile:1(
+  AssignmentStmt:1(
+    NameExpr(x)
+    TempNode:1(
+      Any)
+    tuple?[int?, *Ts?, str?]))

test-data/unit/native-parser-python312.test

@@ -0,0 +1,92 @@
+-- Native parser test cases that target Python 3.12.
+
+[case testPEP695TypeAlias]
+# comment
+type A[T] = C[T]
+[out]
+MypyFile:1(
+  TypeAliasStmt:2(
+    NameExpr(A)
+    TypeParam(
+      T)
+    LambdaExpr:2(
+      Block:-1(
+        ReturnStmt:2(
+          IndexExpr:2(
+            NameExpr(C)
+            NameExpr(T)))))))
+
+[case testPEP695GenericFunction]
+# comment
+
+def f[T](): pass
+def g[T: str](): pass
+def h[T: (int, str)](): pass
+[out]
+MypyFile:1(
+  FuncDef:3(
+    f
+    TypeParam(
+      T)
+    Block:3(
+      PassStmt:3()))
+  FuncDef:4(
+    g
+    TypeParam(
+      T
+      str?)
+    Block:4(
+      PassStmt:4()))
+  FuncDef:5(
+    h
+    TypeParam(
+      T
+      Values(
+        int?
+        str?))
+    Block:5(
+      PassStmt:5())))
+
+[case testPEP695ParamSpec]
+# comment
+
+def f[**P](): pass
+class C[T: int, **P]: pass
+[out]
+MypyFile:1(
+  FuncDef:3(
+    f
+    TypeParam(
+      **P)
+    Block:3(
+      PassStmt:3()))
+  ClassDef:4(
+    C
+    TypeParam(
+      T
+      int?)
+    TypeParam(
+      **P)
+    PassStmt:4()))
+
+[case testPEP695TypeVarTuple]
+# comment
+
+def f[*Ts](): pass
+class C[T: int, *Ts]: pass
+[out]
+MypyFile:1(
+  FuncDef:3(
+    f
+    TypeParam(
+      *Ts)
+    Block:3(
+      PassStmt:3()))
+  ClassDef:4(
+    C
+    TypeParam(
+      T
+      int?)
+    TypeParam(
+      *Ts)
+    PassStmt:4()))