Skip to content

Commit 822813d

Browse files
committed
Duplicate walrus comprehension tests to global scope (now failing)
1 parent c0f633f commit 822813d

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

test-data/unit/check-python38.test

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -201,6 +201,10 @@ while b := "x":
201201
l = [y2 := 1, y2 + 2, y2 + 3]
202202
reveal_type(y2) # N: Revealed type is 'builtins.int'
203203
reveal_type(l) # N: Revealed type is 'builtins.list[builtins.int*]'
204+
205+
filtered_data = [y3 for x in l if (y3 := a) is not None]
206+
reveal_type(filtered_data) # N: Revealed type is 'builtins.list[builtins.int*]'
207+
reveal_type(y3) # N: Revealed type is 'builtins.int'
204208

205209
d = {'a': (a2 := 1), 'b': a2 + 1, 'c': a2 + 2}
206210
reveal_type(d) # N: Revealed type is 'builtins.dict[builtins.str*, builtins.int*]'
@@ -211,6 +215,10 @@ reveal_type(d2) # N: Revealed type is 'builtins.dict[builtins.str*, builtins.in
211215
reveal_type(prefix) # N: Revealed type is 'builtins.str'
212216
reveal_type(start_val) # N: Revealed type is 'builtins.int'
213217

218+
filtered_dict = {k: new_v for k, v in [('a', 1), ('b', 2), ('c', 3)] if (new_v := v + 1) == 2}
219+
reveal_type(filtered_dict) # N: Revealed type is 'builtins.dict[builtins.str*, builtins.int*]'
220+
reveal_type(new_v) # N: Revealed type is 'builtins.int'
221+
214222
def f(x: int = (c := 4)) -> int:
215223
if a := 2:
216224
reveal_type(a) # N: Revealed type is 'builtins.int'

0 commit comments

Comments
 (0)