clarify testReplaceIsDataclass

python · Aug 28, 2023 · 1a538d4 · 1a538d4
1 parent 0b95f3e
commit 1a538d4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/test-data/unit/check-dataclasses.test b/test-data/unit/check-dataclasses.test
@@ -2237,11 +2237,11 @@ replace(None)  # E: Value of type variable "_DataclassT" of "replace" cannot be
 from dataclasses import is_dataclass, replace
 
 def f(x: object) -> None:
-  # error before type-guard
-  y = replace(x)  # E: Value of type variable "_DataclassT" of "replace" cannot be "object"
-  # no error after type-guard
-  if is_dataclass(x) and not isinstance(x, type):
-    y = replace(x)
+  _ = replace(x)  # E: Value of type variable "_DataclassT" of "replace" cannot be "object"
+  if is_dataclass(x):
+    _ = replace(x)  # E: Value of type variable "_DataclassT" of "replace" cannot be "Union[DataclassInstance, Type[DataclassInstance]]"
+    if not isinstance(x, type):
+      _ = replace(x)
 
 [builtins fixtures/tuple.pyi]