Yury Selivanov | e9ad5a5 | 2015-05-09 13:53:57 -0400 | [diff] [blame] | 1 | from __future__ import generator_stop |
| 2 | |
| 3 | import unittest |
| 4 | |
| 5 | |
| 6 | class TestPEP479(unittest.TestCase): |
| 7 | def test_stopiteration_wrapping(self): |
| 8 | def f(): |
| 9 | raise StopIteration |
| 10 | def g(): |
| 11 | yield f() |
| 12 | with self.assertRaisesRegex(RuntimeError, |
| 13 | "generator raised StopIteration"): |
| 14 | next(g()) |
| 15 | |
| 16 | def test_stopiteration_wrapping_context(self): |
| 17 | def f(): |
| 18 | raise StopIteration |
| 19 | def g(): |
| 20 | yield f() |
| 21 | |
| 22 | try: |
| 23 | next(g()) |
| 24 | except RuntimeError as exc: |
| 25 | self.assertIs(type(exc.__cause__), StopIteration) |
| 26 | self.assertIs(type(exc.__context__), StopIteration) |
| 27 | self.assertTrue(exc.__suppress_context__) |
| 28 | else: |
| 29 | self.fail('__cause__, __context__, or __suppress_context__ ' |
| 30 | 'were not properly set') |
| 31 | |
| 32 | |
| 33 | if __name__ == '__main__': |
| 34 | unittest.main() |