1berry/cpython
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

gh-109182: Fix and improve tests for gh-108654 (GH-109189)

Browse Source
This commit is contained in:
Serhiy Storchaka 2023-09-11 17:50:33 +03:00 committed by GitHub
parent 4a69301ea4
commit c0f488b88f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
Lib/test/test_listcomps.py
View File

@ -125,7 +125,7 @@ class ListComprehensionTest(unittest.TestCase):
self.assertIs(type(e), raises)
else:
for k, v in (outputs or {}).items():
self.assertEqual(get_output(newns, k), v)
self.assertEqual(get_output(newns, k), v, k)
def test_lambdas_with_iteration_var_as_default(self):
code = """
@ -563,28 +563,38 @@ class ListComprehensionTest(unittest.TestCase):
def test_comp_in_try_except(self):
template = """
value = ["a"]
value = ["ab"]
result = snapshot = None
try:
[{func}(value) for value in value]
result = [{func}(value) for value in value]
except:
pass
snapshot = value
raise
"""
for func in ["str", "int"]:
code = template.format(func=func)
raises = func != "str"
with self.subTest(raises=raises):
self._check_in_scopes(code, {"value": ["a"]})
# No exception.
code = template.format(func='len')
self._check_in_scopes(code, {"value": ["ab"], "result": [2], "snapshot": None})
# Handles exception.
code = template.format(func='int')
self._check_in_scopes(code, {"value": ["ab"], "result": None, "snapshot": ["ab"]},
raises=ValueError)
def test_comp_in_try_finally(self):
code = """
def f(value):
try:
[{func}(value) for value in value]
finally:
return value
ret = f(["a"])
template = """
value = ["ab"]
result = snapshot = None
try:
result = [{func}(value) for value in value]
finally:
snapshot = value
"""
self._check_in_scopes(code, {"ret": ["a"]})
# No exception.
code = template.format(func='len')
self._check_in_scopes(code, {"value": ["ab"], "result": [2], "snapshot": ["ab"]})
# Handles exception.
code = template.format(func='int')
self._check_in_scopes(code, {"value": ["ab"], "result": None, "snapshot": ["ab"]},
raises=ValueError)
def test_exception_in_post_comp_call(self):
code = """