Staging
v0.5.1
https://github.com/python/cpython
Revision efcf82f94572abcdbd70336e0b2c3d0f4df280bc authored by Serhiy Storchaka on 15 January 2019, 08:53:18 UTC, committed by GitHub on 15 January 2019, 08:53:18 UTC
1 parent 6fe9c44
Raw File
Tip revision: efcf82f94572abcdbd70336e0b2c3d0f4df280bc authored by Serhiy Storchaka on 15 January 2019, 08:53:18 UTC
bpo-35619: Improve support of custom data descriptors in help() and pydoc. (GH-11366)
Tip revision: efcf82f
test_generator_stop.py
from __future__ import generator_stop

import unittest


class TestPEP479(unittest.TestCase):
    def test_stopiteration_wrapping(self):
        def f():
            raise StopIteration
        def g():
            yield f()
        with self.assertRaisesRegex(RuntimeError,
                                    "generator raised StopIteration"):
            next(g())

    def test_stopiteration_wrapping_context(self):
        def f():
            raise StopIteration
        def g():
            yield f()

        try:
            next(g())
        except RuntimeError as exc:
            self.assertIs(type(exc.__cause__), StopIteration)
            self.assertIs(type(exc.__context__), StopIteration)
            self.assertTrue(exc.__suppress_context__)
        else:
            self.fail('__cause__, __context__, or __suppress_context__ '
                      'were not properly set')


if __name__ == '__main__':
    unittest.main()
back to top