From c553a74741e553bf96cf6685de4cac65c3fface4 Mon Sep 17 00:00:00 2001
From: popsiclelmlm <7487674+popsiclelmlm@users.noreply.github.com>
Date: Thu, 11 Jun 2026 02:41:29 +0800
Subject: [PATCH] Add walrus redefinition test cases

---
 test-data/unit/check-redefine2.test | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/test-data/unit/check-redefine2.test b/test-data/unit/check-redefine2.test
index 2af6959dd21a6..9b5deeeb541ba 100644
--- a/test-data/unit/check-redefine2.test
+++ b/test-data/unit/check-redefine2.test
@@ -333,6 +333,18 @@ def f3() -> None:
         reveal_type(x) # N: Revealed type is "builtins.int | builtins.str"
     reveal_type(x) # N: Revealed type is "builtins.int | builtins.str"
 
+def f4(x: str | None) -> None:
+    if y := x:
+        reveal_type(y) # N: Revealed type is "builtins.str"
+    y = 1
+    reveal_type(y) # N: Revealed type is "builtins.int"
+
+def f5() -> None:
+    if z := 1:
+        reveal_type(z) # N: Revealed type is "builtins.int"
+    z = "s"
+    reveal_type(z) # N: Revealed type is "builtins.str"
+
 [case testNewRedefineOperatorAssignment]
 # flags: --allow-redefinition
 class D: pass