Commit fafd874b authored by Jim Fulton's avatar Jim Fulton

Added an __iter__ method for test suites.

parent 9f556a40
"""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()
......@@ -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:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment