From 37d89a295540bb56e230a4935912f0fe5bdd0301 Mon Sep 17 00:00:00 2001 From: Michael Foord Date: Sun, 5 Apr 2009 01:15:01 +0000 Subject: Change the way unittest.TestSuite use their tests to always access them through iteration. Non behavior changing, this allows you to create custom subclasses that override __iter__. Issue #5693 --- Lib/unittest.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/Lib/unittest.py b/Lib/unittest.py index 3eedd56..83790fc 100644 --- a/Lib/unittest.py +++ b/Lib/unittest.py @@ -998,7 +998,7 @@ class TestSuite(object): self.addTests(tests) def __repr__(self): - return "<%s tests=%s>" % (_strclass(self.__class__), self._tests) + return "<%s tests=%s>" % (_strclass(self.__class__), list(self)) def __eq__(self, other): if not isinstance(other, self.__class__): @@ -1016,7 +1016,7 @@ class TestSuite(object): def countTestCases(self): cases = 0 - for test in self._tests: + for test in self: cases += test.countTestCases() return cases @@ -1036,7 +1036,7 @@ class TestSuite(object): self.addTest(test) def run(self, result): - for test in self._tests: + for test in self: if result.shouldStop: break test(result) @@ -1047,7 +1047,7 @@ class TestSuite(object): def debug(self): """Run the tests without collecting errors in a TestResult""" - for test in self._tests: + for test in self: test.debug() -- cgit v0.12