add test case for overriding defaults

python · JukkaL · Feb 8, 2023 · Jan 30, 2023 · Feb 1, 2023 · Feb 3, 2023
commit 91772caeb6a3a627e55f877df03bff3ac1a5a75d
diff --git a/test-data/unit/check-dataclass-transform.test b/test-data/unit/check-dataclass-transform.test
@@ -162,6 +162,27 @@ b.x = 2  # E: Property "x" defined in "B" is read-only
 [typing fixtures/typing-full.pyi]
 [builtins fixtures/dataclasses.pyi]
 
+[case testDataclassTransformDefaultsCanBeOverridden]
+# flags: --python-version 3.11
+from typing import dataclass_transform, Callable, Type
+
+@dataclass_transform(kw_only_default=True)
+def my_dataclass(*, kw_only: bool = True) -> Callable[[Type], Type]:
+    return lambda cls: cls
+
+@my_dataclass()
+class KwOnly:
+    x: int
+@my_dataclass(kw_only=False)
+class KwOptional:
+    x: int
+
+KwOnly(5)  # E: Too many positional arguments for "KwOnly"
+KwOptional(5)
+
+[typing fixtures/typing-full.pyi]
+[builtins fixtures/dataclasses.pyi]
+
 [case testDataclassTransformFieldSpecifiersDefaultsToEmpty]
 # flags: --python-version 3.11
 from dataclasses import field, dataclass