Support patterns in module names

kvalev · kvalev · commit 44e2fec038fb · 2021-01-19T11:10:38.000+02:00
diff --git a/coverage/files.py b/coverage/files.py
@@ -242,6 +242,7 @@ class ModuleMatcher(object):
     """A matcher for modules in a tree."""
     def __init__(self, module_names):
         self.modules = list(module_names)
+        self.module_patterns = [re.compile(m) for m in self.modules]
 
     def __repr__(self):
         return "<ModuleMatcher %r>" % (self.modules)
@@ -263,6 +264,10 @@ def match(self, module_name):
                     # This is a module in the package
                     return True
 
+        for pattern in self.module_patterns:
+            if pattern.match(module_name):
+                return True
+
         return False
 
 
diff --git a/tests/test_files.py b/tests/test_files.py
@@ -174,12 +174,15 @@ def test_tree_matcher(self):
     def test_module_matcher(self):
         matches_to_try = [
             ('test', True),
+            ('te.*', True),
             ('trash', False),
+            ('tra.*', False),
             ('testing', False),
             ('test.x', True),
             ('test.x.y.z', True),
             ('py', False),
             ('py.t', False),
+            ('py.t.*', True),
             ('py.test', True),
             ('py.testing', False),
             ('py.test.buz', True),
