diff --git a/python 2/koans/about_generators.py b/python 2/koans/about_generators.py index 9e77ed061..9cc11212a 100644 --- a/python 2/koans/about_generators.py +++ b/python 2/koans/about_generators.py @@ -104,7 +104,15 @@ def test_generators_can_take_coroutines(self): next(generator) self.assertEqual(__, generator.send(1 + 2)) - + + def test_before_sending_a_value_to_a_generator_next_must_be_called(self): + generator = self.generator_with_coroutine() + + try: + generator.send(1+2) + except TypeError as ex: + self.assertMatch(__, ex[0]) + # ------------------------------------------------------------------ def yield_tester(self): diff --git a/python 3/koans/about_generators.py b/python 3/koans/about_generators.py index 28a148abb..e22b677c7 100644 --- a/python 3/koans/about_generators.py +++ b/python 3/koans/about_generators.py @@ -107,6 +107,14 @@ def test_generators_can_take_coroutines(self): next(generator) self.assertEqual(__, generator.send(1 + 2)) + + def test_before_sending_a_value_to_a_generator_next_must_be_called(self): + generator = self.generator_with_coroutine() + + try: + generator.send(1+2) + except TypeError as ex: + self.assertMatch(__, ex[0]) # ------------------------------------------------------------------