Fix variant constructor type in parser

hazelgrove · Nov 4, 2024 · ac4e9f9 · ac4e9f9
1 parent c3de67f
commit ac4e9f9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/haz3lmenhir/Parser.mly b/src/haz3lmenhir/Parser.mly
@@ -244,7 +244,8 @@ exp:
     | f = FLOAT { Float f }
     | v = IDENT { Var v }
     | c = CONSTRUCTOR_IDENT { Constructor(c, UnknownType(Internal))}
-    | c = CONSTRUCTOR_IDENT; TILDE; t = typ;  { Cast(Constructor(c, UnknownType(Internal)), UnknownType(Internal), t) }
+    | c = CONSTRUCTOR_IDENT; TILDE; t = typ;  { Constructor(c, t) }
+    | c = CONSTRUCTOR_IDENT; COLON; t = typ;  { Cast(Constructor(c, UnknownType(Internal)), UnknownType(Internal), t) }
     | s = STRING { String s}
     | OPEN_PAREN; e = exp; CLOSE_PAREN { e } 
     | OPEN_PAREN; e = exp; COMMA; l = separated_list(COMMA, exp); CLOSE_PAREN { TupleExp(e :: l) }

diff --git a/test/Test_Menhir.re b/test/Test_Menhir.re
@@ -266,6 +266,11 @@ let tests = [
       Int |> Typ.fresh,
     )
     |> Exp.fresh,
+    "A : Int",
+  ),
+  menhir_only_test(
+    "Constructor of specific sum type",
+    Constructor("A", Int |> Typ.fresh) |> Exp.fresh,
     "A ~ Int",
   ),
   parser_test(