diff options
Diffstat (limited to 'coverage/control.py')
| -rw-r--r-- | coverage/control.py | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/coverage/control.py b/coverage/control.py index 63c9d382..1044fad0 100644 --- a/coverage/control.py +++ b/coverage/control.py @@ -308,11 +308,18 @@ class Coverage(object): def _source_for_file(self, filename): """Return the source file for `filename`.""" - if not filename.endswith(".py"): - if filename[-4:-1] == ".py": - filename = filename[:-1] - elif filename.endswith("$py.class"): # jython - filename = filename[:-9] + ".py" + if filename.endswith(".py"): + return filename + elif filename.endswith((".pyc", ".pyo")): + try_filename = filename[:-1] + if os.path.exists(try_filename): + return try_filename + if sys.platform == "win32": + try_filename += "w" + if os.path.exists(try_filename): + return try_filename + elif filename.endswith("$py.class"): # Jython + filename = filename[:-9] + ".py" return filename def _name_for_module(self, module_globals, filename): |