From 2b304497bbad3a4a4d84f719a17ea080e4bbf708 Mon Sep 17 00:00:00 2001 From: loic Date: Tue, 10 Jan 2017 22:21:13 +0100 Subject: use the new source_for_file helper where it makes sense --- 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 376d3f5..7969432 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