Add tests for weakref support for generator-iterators.
Part of fixing SF bug #591704.
diff --git a/Lib/test/test_generators.py b/Lib/test/test_generators.py
index 170d92e..9078b55 100644
--- a/Lib/test/test_generators.py
+++ b/Lib/test/test_generators.py
@@ -1359,12 +1359,38 @@
+---+---+---+---+---+---+---+---+---+---+
"""
+weakref_tests = """\
+Generators are weakly referencable:
+
+>>> import weakref
+>>> def gen():
+... yield 'foo!'
+...
+>>> wr = weakref.ref(gen)
+>>> wr() is gen
+True
+>>> p = weakref.proxy(gen)
+
+Generator-iterators are weakly referencable as well:
+
+>>> gi = gen()
+>>> wr = weakref.ref(gi)
+>>> wr() is gi
+True
+>>> p = weakref.proxy(gi)
+>>> list(p)
+['foo!']
+
+"""
+
__test__ = {"tut": tutorial_tests,
"pep": pep_tests,
"email": email_tests,
"fun": fun_tests,
"syntax": syntax_tests,
- "conjoin": conjoin_tests}
+ "conjoin": conjoin_tests,
+ "weakref": weakref_tests,
+ }
# Magic test name that regrtest.py invokes *after* importing this module.
# This worms around a bootstrap problem.