diff options
Diffstat (limited to 'Lib/test/test_setcomps.py')
| -rw-r--r-- | Lib/test/test_setcomps.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/Lib/test/test_setcomps.py b/Lib/test/test_setcomps.py index ba4173c..0bb02ef 100644 --- a/Lib/test/test_setcomps.py +++ b/Lib/test/test_setcomps.py @@ -168,8 +168,15 @@ class SetComprehensionTest(unittest.TestCase): except Exception as e: return e + def iter_raises(): + try: + {x for x in BrokenIter(iter_raises=True)} + except Exception as e: + return e + for func, expected in [(init_raises, "BrokenIter(init_raises=True)"), (next_raises, "BrokenIter(next_raises=True)"), + (iter_raises, "BrokenIter(iter_raises=True)"), ]: with self.subTest(func): exc = func() |