From fafd874bc8ae89a9c145e38d94cb14e628fcad34 Mon Sep 17 00:00:00 2001
From: Jim Fulton <jim@zope.com>
Date: Sat, 28 Aug 2004 15:22:12 +0000
Subject: Added an __iter__ method for test suites.

---
 Lib/test/test_unittest.py | 31 +++++++++++++++++++++++++++++++
 Lib/unittest.py           |  3 +++
 2 files changed, 34 insertions(+)
 create mode 100644 Lib/test/test_unittest.py

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:
-- 
cgit v0.12