diff --git a/IPython/testing/iptest.py b/IPython/testing/iptest.py index 6981827..59c3495 100644 --- a/IPython/testing/iptest.py +++ b/IPython/testing/iptest.py @@ -267,6 +267,8 @@ class IPTester(object): call_args = None #: list, process ids of subprocesses we start (for cleanup) pids = None + #: str, coverage xml output file + coverage_xml = None def __init__(self, runner='iptest', params=None): """Create new test runner.""" @@ -285,9 +287,14 @@ class IPTester(object): # Assemble call self.call_args = self.runner+self.params + sect = [p for p in self.params if p.startswith('IPython')][0] if '--with-xunit' in self.call_args: - sect = [p for p in self.params if p.startswith('IPython')][0] self.call_args.append('--xunit-file=%s' % path.abspath(sect+'.xunit.xml')) + + if '--with-coverage' in self.call_args: + self.coverage_xml = path.abspath(sect+".coverage.xml") + self.call_args.remove('--with-coverage') + self.call_args = ["python-coverage", "run", "--source="+sect] + self.call_args[1:] # Store pids of anything we start to clean up on deletion, if possible # (on posix only, since win32 has no os.kill) @@ -319,11 +326,15 @@ class IPTester(object): def run(self): """Run the stored commands""" try: - return self._run_cmd() + retcode = self._run_cmd() except: import traceback traceback.print_exc() return 1 # signal failure + + if self.coverage_xml: + subprocess.check_call(["python-coverage", "xml", "-o", self.coverage_xml]) + return retcode def __del__(self): """Cleanup on exit by killing any leftover processes."""