diff options
Diffstat (limited to 'Lib/lib2to3/tests')
-rwxr-xr-x | Lib/lib2to3/tests/test_fixers.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Lib/lib2to3/tests/test_fixers.py b/Lib/lib2to3/tests/test_fixers.py index eabfae5..f8cdfb9 100755 --- a/Lib/lib2to3/tests/test_fixers.py +++ b/Lib/lib2to3/tests/test_fixers.py @@ -33,8 +33,10 @@ class FixerTestCase(support.TestCase): self.fixer_log = [] self.filename = "<string>" - for order in (self.refactor.pre_order, self.refactor.post_order): - for fixer in order: + from itertools import chain + for order in (self.refactor.pre_order.values(),\ + self.refactor.post_order.values()): + for fixer in chain(*order): fixer.log = self.fixer_log def _check(self, before, after): |