Match, match_case, and MatchValue

src/latexify/codegen/function_codegen.py

@@ -536,6 +536,23 @@ def visit_IfExp(self, node: ast.IfExp) -> str:
         latex += self.visit(node)
         return latex + r", & \mathrm{otherwise} \end{array} \right."
 
+    def visit_Match(self, node: ast.Match) -> str:
+        """Visit a match node"""
+        latex = r"\left\{ \begin{array}{ll} "
+        subject_latex = self.visit(node.subject)        
+        for i, match_case in enumerate(node.cases):
+            true_latex = self.visit(match_case.body[0])
+            cond_latex = self.visit(match_case.pattern)
+            latex += true_latex + r", & \mathrm{if} \ " + subject_latex + cond_latex + r" \\ "
+
+        latex += r"\end{array} \right."
+        return latex
+
+    def visit_MatchValue(self, node: ast.MatchValue) -> str:
+        """Visit a MatchValue node"""
+        latex = self.visit(node.value)
+        return r" = " + latex
+
     def _reduce_stop_parameter(self, node: ast.BinOp) -> str:
         # ast.Constant class is added in Python 3.8
         # ast.Num is the relevant node type in previous versions

src/latexify/codegen/function_codegen_test.py

@@ -10,7 +10,6 @@
 from latexify import exceptions, test_utils
 from latexify.codegen import FunctionCodegen, function_codegen
 
-
 def test_generic_visit() -> None:
     class UnknownNode(ast.AST):
         pass
@@ -744,3 +743,35 @@ def test_use_set_symbols_compare(code: str, latex: str) -> None:
     tree = ast.parse(code).body[0].value
     assert isinstance(tree, ast.Compare)
     assert function_codegen.FunctionCodegen(use_set_symbols=True).visit(tree) == latex
+
+
+def test_matchvalue() -> None:
+    tree = ast.parse(
+        textwrap.dedent(
+        """
+        match x:
+            case 0:
+                return 1
+        """
+        )
+    ).body[0]
+
+    assert FunctionCodegen().visit(tree) == r"\left\{ \begin{array}{ll} {1}, & \mathrm{if} \ x = {0} \\ \end{array} \right."
+
+
+def test_multiple_matchvalue() -> None:
+    tree = ast.parse(
+        textwrap.dedent(
+        """
+        match x:
+            case 0:
+                return 1
+            case 1:
+                return 2
+        """
+        )
+    ).body[0]
+
+    assert FunctionCodegen().visit(tree) == r"\left\{ \begin{array}{ll} {1}, & \mathrm{if} \ x = {0} \\ {2}, & \mathrm{if} \ x = {1} \\ \end{array} \right."
+
+