From: Jim Fulton <jim@zope.com>
Date: Sat, 28 Aug 2004 15:22:12 +0000 (+0000)
Subject: Added an __iter__ method for test suites.
X-Git-Tag: v2.4a3~77
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=fafd874bc8ae89a9c145e38d94cb14e628fcad34;p=python

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 0000000000..91511667a6
--- /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 29d90e3bfa..3375067c8d 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: