Allow regexes on specific messages.

Author: brunns

README.md

@@ -139,6 +139,15 @@ Implementation notes:
     main:2: note: .*str.*
 ```
 
+#### 5. Regular expressions specific lines of output.
+
+```yaml
+- case: expected_single_message_regex
+  main: |
+    a = 'hello'
+    reveal_type(a)  # NR: .*str.*
+```
+
 ## Options
 
 ```

pytest_mypy_plugins/tests/test-regex_assertions.yml

@@ -14,3 +14,9 @@
     reveal_type(a)
   out: |
     main:2: note: .*str.*
+
+- case: expected_single_message_regex
+  regex: no
+  main: |
+    a = 'hello'
+    reveal_type(a)  # NR: .*str.*

pytest_mypy_plugins/utils.py

@@ -302,29 +302,42 @@ def extract_output_matchers_from_comments(fname: str, input_lines: List[str], re
     """Transform comments such as '# E: message' or
     '# E:3: message' in input.
 
-    The result is lines like 'fnam:line: error: message'.
+    The result is a list pf output matchers
     """
     fname = fname.replace(".py", "")
     matchers = []
     for index, line in enumerate(input_lines):
         # The first in the split things isn't a comment
         for possible_err_comment in line.split(" # ")[1:]:
-            m = re.search(r"^([ENW]):((?P<col>\d+):)? (?P<message>.*)$", possible_err_comment.strip())
-            if m:
-                if m.group(1) == "E":
+            match = re.search(r"^([ENW])(?P<regex>[R]):((?P<col>\d+):)? (?P<message>.*)$", possible_err_comment.strip())
+            if match:
+                if match.group(1) == "E":
                     severity = "error"
-                elif m.group(1) == "N":
+                elif match.group(1) == "N":
                     severity = "note"
-                elif m.group(1) == "W":
+                elif match.group(1) == "W":
                     severity = "warning"
-                col = m.group("col")
+                else:
+                    severity = match.group(1)
+                col = match.group("col")
                 matchers.append(
-                    OutputMatcher(fname, index + 1, severity, message=m.group("message"), regex=regex, col=col)
+                    OutputMatcher(
+                        fname,
+                        index + 1,
+                        severity,
+                        message=match.group("message"),
+                        regex=regex or bool(match.group("regex")),
+                        col=col,
+                    )
                 )
     return matchers
 
 
 def extract_output_matchers_from_out(out: str, params: Mapping[str, Any], regex: bool) -> List[OutputMatcher]:
+    """Transform output lines such as 'function:9: E: message'
+
+    The result is a list of output matchers
+    """
     matchers = []
     for line in pystache.render(out, params).split("\n"):
         match = re.search(