From d89ff759ec85b134d21518cc6693fa4df7808682 Mon Sep 17 00:00:00 2001 From: Tobias Schmidt <tobias.schmidt@unibas.ch> Date: Fri, 27 Apr 2012 12:12:48 +0200 Subject: [PATCH] remove with statement in xmlrunner --- modules/base/pymod/xmlrunner.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/modules/base/pymod/xmlrunner.py b/modules/base/pymod/xmlrunner.py index 86c10b3f9..e2bcae76d 100644 --- a/modules/base/pymod/xmlrunner.py +++ b/modules/base/pymod/xmlrunner.py @@ -5,8 +5,6 @@ XML Test Runner for PyUnit # Written by Sebastian Rittau <srittau@jroger.in-berlin.de> and placed in # the Public Domain. With contributions by Paolo Borelli and others. -from __future__ import with_statement - __version__ = "0.1" import os.path @@ -184,7 +182,11 @@ class XMLTestRunner(object): result = _XMLTestResult(classname) start_time = time.time() - with _fake_std_streams(): + try: + self._orig_stdout = sys.stdout + self._orig_stderr = sys.stderr + sys.stdout = StringIO() + sys.stderr = StringIO() test(result) try: out_s = sys.stdout.getvalue() @@ -194,6 +196,10 @@ class XMLTestRunner(object): err_s = sys.stderr.getvalue() except AttributeError: err_s = "" + finally: + sys.stdout = self._orig_stdout + sys.stderr = self._orig_stderr + time_taken = time.time() - start_time result.print_report(stream, time_taken, out_s, err_s) -- GitLab