Added an __iter__ method for test suites.
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 @@
__str__ = __repr__
+ def __iter__(self):
+ return iter(self._tests)
+
def countTestCases(self):
cases = 0
for test in self._tests: