Fix number type constraint to allow types with methods. (#3637)

savonet · Jan 15, 2024 · 2f7fb67 · 2f7fb67
1 parent 008f275
commit 2f7fb67
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/lang/type.ml b/src/lang/type.ml
@@ -45,9 +45,10 @@ let num_constr =
     constr_descr = "a number type";
     univ_descr = None;
     satisfied =
-      (fun ~subtype:_ ~satisfies:_ b ->
+      (fun ~subtype:_ ~satisfies b ->
         let b = demeth b in
         match b.descr with
+          | Var _ -> satisfies b
           | Custom { typ = Ground.Never.Type }
           | Custom { typ = Ground.Int.Type }
           | Custom { typ = Ground.Float.Type } ->

diff --git a/tests/language/number.liq b/tests/language/number.liq
@@ -48,6 +48,16 @@ def f() =
   test.equals(round(0.), 0.)
   test.equals(round(infinity), infinity)
   test.equals(round(nan), nan)
+
+  def f(x) =
+    print(
+      "x.foo is: " ^
+        x.foo
+    )
+    x + 1
+  end
+  test.equals(f(1.{foo="gni"}), 2)
+
   test.pass()
 end