gh-104602: Add additional test for listcomp with lambda (#104639)

This threw a SystemError before #104603. Adding a separate test
because this was a different failure mode than the other two new
tests from #104603, both of which used to segfault.
This commit is contained in:
Jelle Zijlstra 2023-05-19 08:16:39 -07:00 committed by GitHub
parent 8a8853af24
commit dbe171e609
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -510,6 +510,16 @@ class ListComprehensionTest(unittest.TestCase):
"""
self._check_in_scopes(code, {"z": 1, "out": [(3, 2, 1)]})
def test_lambda_in_iter(self):
code = """
(func, c), = [(a, b) for b in [1] for a in [lambda : a]]
d = func()
assert d is func
# must use "a" in this scope
e = a if False else None
"""
self._check_in_scopes(code, {"c": 1, "e": None})
def test_assign_to_comp_iter_var_in_outer_function(self):
code = """
a = [1 for a in [0]]