Made it possible to just "yield test" in test generators

1 files changed, 21 insertions, 0 deletions

diff --git a/unit_tests/test_issue_230.py b/unit_tests/test_issue_230.py
new file mode 100644
index 0000000..41a717b
--- /dev/null
+++ b/unit_tests/test_issue_230.py
@@ -0,0 +1,21 @@
+import os
+import unittest
+
+class TestIssue230(unittest.TestCase):
+
+    def test_generator_yield_value(self):
+        from nose.loader import TestLoader
+
+        def test():
+            pass
+        def gen():
+            yield test
+
+        loader = TestLoader()
+        suite = loader.loadTestsFromGenerator(gen, module=None)
+        testcase = iter(suite).next()
+        self.assertEqual(testcase.test.test, test)
+
+
+if __name__ == '__main__':
+    unittest.main()