[match case] Use match case in assertrepr_compare

Author: Pierre-Sassoulas

src/_pytest/assertion/util.py

@@ -43,6 +43,15 @@ def __call__(self, source: str, lexer: Literal["diff", "python"] = "python") ->
         """Apply highlighting to the given source."""
 
 
+CompareSetFunction = dict[
+    str,
+    Callable[
+        [AbstractSet[Any], AbstractSet[Any], _HighlightFunc, int],
+        list[str],
+    ],
+]
+
+
 def dummy_highlighter(source: str, lexer: Literal["diff", "python"] = "python") -> str:
     """Dummy highlighter that returns the text unprocessed.
 
@@ -204,30 +213,28 @@ def assertrepr_compare(
 
     summary = f"{left_repr} {op} {right_repr}"
     highlighter = config.get_terminal_writer()._highlight
-
-    explanation = None
+    explanation: list[str] | None
     try:
-        if op == "==":
-            explanation = _compare_eq_any(left, right, highlighter, verbose)
-        elif op == "not in":
-            if istext(left) and istext(right):
+        match (left, op, right):
+            case (_, "==", _):
+                explanation = _compare_eq_any(left, right, highlighter, verbose)
+            case (str(), "not in", str()):
                 explanation = _notin_text(left, right, verbose)
-        elif op == "!=":
-            if isset(left) and isset(right):
-                explanation = ["Both sets are equal"]
-        elif op == ">=":
-            if isset(left) and isset(right):
-                explanation = _compare_gte_set(left, right, highlighter, verbose)
-        elif op == "<=":
-            if isset(left) and isset(right):
-                explanation = _compare_lte_set(left, right, highlighter, verbose)
-        elif op == ">":
-            if isset(left) and isset(right):
-                explanation = _compare_gt_set(left, right, highlighter, verbose)
-        elif op == "<":
-            if isset(left) and isset(right):
-                explanation = _compare_lt_set(left, right, highlighter, verbose)
-
+            case (
+                set() | frozenset(),
+                "!=" | ">=" | "<=" | ">" | "<",
+                set() | frozenset(),
+            ):
+                set_compare_func: CompareSetFunction = {
+                    "!=": lambda *a, **kw: ["Both sets are equal"],
+                    ">=": _compare_gte_set,
+                    "<=": _compare_lte_set,
+                    ">": _compare_gt_set,
+                    "<": _compare_lt_set,
+                }
+                explanation = set_compare_func[op](left, right, highlighter, verbose)
+            case _:
+                explanation = None
     except outcomes.Exit:
         raise
     except Exception:

testing/test_assertion.py

@@ -1976,10 +1976,10 @@ def f():
 
 
 def test_exit_from_assertrepr_compare(monkeypatch) -> None:
-    def raise_exit(obj):
+    def raise_exit(*args, **kwargs):
         outcomes.exit("Quitting debugger")
 
-    monkeypatch.setattr(util, "istext", raise_exit)
+    monkeypatch.setattr(util, "_compare_eq_any", raise_exit)
 
     with pytest.raises(outcomes.Exit, match="Quitting debugger"):
         callequal(1, 1)