From 2d1e9ba1072951a2c0eb8d280520fbc40b571652 Mon Sep 17 00:00:00 2001
From: Michele Simionato <michele.simionato@gmail.com>
Date: Sun, 4 Nov 2018 06:17:59 +0100
Subject: Added a test for decorating generators

---
 src/tests/test.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'src/tests')

diff --git a/src/tests/test.py b/src/tests/test.py
index 3c4dcd1..9068b11 100644
--- a/src/tests/test.py
+++ b/src/tests/test.py
@@ -58,6 +58,23 @@ class CoroutineTestCase(unittest.TestCase):
 ''')
 
 
+def gen123():
+    yield 1
+    yield 2
+    yield 3
+
+
+class GeneratorCallerTestCase(unittest.TestCase):
+    def test_gen123(self):
+        @decorator
+        def square(func, *args, **kw):
+            for x in gen123():
+                yield x * x
+        new = square(gen123)
+        self.assertTrue(inspect.isgeneratorfunction(new))
+        self.assertEqual(list(new()), [1, 4, 9])
+
+
 class DocumentationTestCase(unittest.TestCase):
     def test(self):
         err = doctest.testmod(doc)[0]
-- 
cgit v1.2.1