Added test demonstrating that calling send on a generator with a non-None argument is illegal.

icmurray · icmurray · commit bfdebc5b02c7 · 2011-02-04T13:06:41.000Z
Demonstrates a TypeError is raised.
diff --git 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
@@ -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])
                 
     # ------------------------------------------------------------------
     
