from __future__ import generator_stop

import unittest

class TestPEP479(unittest.TestCase):
    def test_refactored_stopiteration(self):
        def f():
            raise StopIteration

        def g():
            yield f()

        self.assertRaises(RuntimeError, next, g())