Added an __iter__ method for test suites.

2 files changed, 34 insertions, 0 deletions

diff --git a/Lib/test/test_unittest.py b/Lib/test/test_unittest.py
new file mode 100644
index 0000000..9151166
--- /dev/null
+++ b/Lib/test/test_unittest.py
@@ -0,0 +1,31 @@
+"""Test script for unittest.
+
+This just includes tests for new features.  We really need a
+full set of tests.
+"""
+
+import unittest
+
+def test_TestSuite_iter():
+    """
+    >>> test1 = unittest.FunctionTestCase(lambda: None)
+    >>> test2 = unittest.FunctionTestCase(lambda: None)
+    >>> suite = unittest.TestSuite((test1, test2))
+    >>> tests = []
+    >>> for test in suite:
+    ...     tests.append(test)
+    >>> tests == [test1, test2]
+    True
+    """
+
+
+######################################################################
+## Main
+######################################################################
+
+def test_main():
+    from test import test_support, test_unittest
+    test_support.run_doctest(test_unittest, verbosity=True)
+
+if __name__ == '__main__':
+    test_main()
diff --git a/Lib/unittest.py b/Lib/unittest.py
index 29d90e3..3375067 100644
--- a/Lib/unittest.py
+++ b/Lib/unittest.py
@@ -400,6 +400,9 @@ class TestSuite:
 
     __str__ = __repr__
 
+    def __iter__(self):
+        return iter(self._tests)
+
     def countTestCases(self):
         cases = 0
         for test in self._tests: