From 37c3ddda90d48ddf793d78873bd681177f345d4a Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Tue, 15 Jul 2008 00:27:38 +0300 Subject: Remove directory path from filenames in the output, if it is the current working directory. --- CoverageTestRunner.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'CoverageTestRunner.py') diff --git a/CoverageTestRunner.py b/CoverageTestRunner.py index 7a6370a..5bbcb59 100644 --- a/CoverageTestRunner.py +++ b/CoverageTestRunner.py @@ -71,6 +71,7 @@ class CoverageTestRunner: """A test runner class that insists modules' tests cover them fully.""" def __init__(self): + self._dirname = None self._module_pairs = [] def add_pair(self, module_pathname, test_module_pathname): @@ -87,7 +88,10 @@ class CoverageTestRunner: """ suffixes = ["_tests.py", "Tests.py"] - + + self._dirname = os.path.abspath(dirname) + if not self._dirname.endswith(os.sep): + self._dirname += os.sep for dirname, dirnames, filenames in os.walk(dirname): tests = [] @@ -143,6 +147,8 @@ class CoverageTestRunner: coverage.stop() filename, stmts, missed, missed_desc = coverage.analysis(module) if missed: + if self._dirname and filename.startswith(self._dirname): + filename = filename[len(self._dirname):] result.addCoverageMissed(filename, stmts, missed, missed_desc) end_time = time.time() -- cgit v1.2.1