From ffc050e243b4c329f582372dd49c55aa225197b9 Mon Sep 17 00:00:00 2001 From: "loic@dachary.org" Date: Tue, 10 Jan 2017 22:21:13 +0100 Subject: use the new source_for_file helper where it makes sense --HG-- branch : issue-426 --- coverage/control.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'coverage/control.py') diff --git a/coverage/control.py b/coverage/control.py index 376d3f54..79694320 100644 --- a/coverage/control.py +++ b/coverage/control.py @@ -810,8 +810,8 @@ class Coverage(object): not hasattr(sys.modules[pkg], '__file__') or not os.path.exists(sys.modules[pkg].__file__)): continue - pkg_file = sys.modules[pkg].__file__ - if not pkg_file.endswith(('__init__.py', '__init__.pyc', '__init__.pyo')): + pkg_file = source_for_file(sys.modules[pkg].__file__) + if not pkg_file.endswith('__init__.py'): continue src_directories.append(self._canonical_dir(sys.modules[pkg])) -- cgit v1.2.1