summaryrefslogtreecommitdiff
path: root/coverage/control.py
diff options
context:
space:
mode:
authorloic <loic@dachary.org>2017-01-10 22:21:13 +0100
committerloic <loic@dachary.org>2017-01-10 22:21:13 +0100
commit2b304497bbad3a4a4d84f719a17ea080e4bbf708 (patch)
treeae821fd46b198e2eda191f6076fdab63678fe7ca /coverage/control.py
parent5ffdccc8a588921cdaef8c3a55ecfb5ff9170309 (diff)
downloadpython-coveragepy-2b304497bbad3a4a4d84f719a17ea080e4bbf708.tar.gz
use the new source_for_file helper where it makes sense
Diffstat (limited to 'coverage/control.py')
-rw-r--r--coverage/control.py4
1 files changed, 2 insertions, 2 deletions
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]))