From 0b79941f37f6bd9a56afac0178713d082a0db0fb Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Wed, 15 Apr 2009 13:22:51 -0400 Subject: If a module was loaded from a .pyc, report its file as .py --- coverage/codeunit.py | 3 +++ coverage/control.py | 3 --- 2 files changed, 3 insertions(+), 3 deletions(-) (limited to 'coverage') diff --git a/coverage/codeunit.py b/coverage/codeunit.py index 221d677..96ed43b 100644 --- a/coverage/codeunit.py +++ b/coverage/codeunit.py @@ -57,6 +57,9 @@ class CodeUnit: f = morf.__file__ else: f = morf + # .pyc files should always refer to a .py instead. + if f.endswith('.pyc'): + f = f[:-1] self.filename = file_locator.canonical_filename(f) if hasattr(morf, '__name__'): diff --git a/coverage/control.py b/coverage/control.py index db9b0a6..cb13f2f 100644 --- a/coverage/control.py +++ b/coverage/control.py @@ -119,9 +119,6 @@ class coverage: filename = code_unit.filename ext = os.path.splitext(filename)[1] source = None - if ext == '.pyc': - filename = filename[:-1] - ext = '.py' if ext == '.py': if not os.path.exists(filename): source = self.file_locator.get_zip_data(filename) -- cgit v1.2.1