diff --git a/Lib/test/test_unittest.py b/Lib/test/test_unittest.py new file mode 100644 index 00000000000..91511667a68 --- /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 29d90e3bfa9..3375067c8d0 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: