diff options
Diffstat (limited to 'coverage/testplugin.py')
| -rw-r--r-- | coverage/testplugin.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/coverage/testplugin.py b/coverage/testplugin.py index f6857586..c93b8599 100644 --- a/coverage/testplugin.py +++ b/coverage/testplugin.py @@ -54,12 +54,8 @@ class CoverageTestWrapper(object): self.coverage.stop() self.coverage.save() - modules = [] - if self.coverPackages: - for name, module in sys.modules.items(): - for package in self.coverPackages: - if module is not None and name.startswith(package): - modules.append(module) + modules = [module for name, module in sys.modules.items() + if self._want_module(name, module)] # Remaining actions are reporting, with some common self.options. report_args = { @@ -95,6 +91,14 @@ class CoverageTestWrapper(object): return + def _want_module(self, name, module): + for package in self.coverPackages: + if module is not None and name.startswith(package): + return True + + return False + + options = [ optparse.Option('', '--cover-action', action='append', default=['report'], |