annotation

proud/basic_impl.py

@@ -379,7 +379,10 @@ def binary(module, head, tl: typing.Tuple[tuple, ...]):
         raise NotImplementedError
 
     def annotate(module, var, type):
-        raise NotImplementedError
+        var = module.eval(var)
+        type = Typing(module.comp_ctx).eval(type)
+        module.comp_ctx.tc_state.unify(var.type, type)
+        return var
 
     def match(module, target, cases: list):
         raise NotImplementedError

test/annotate.py

@@ -0,0 +1,21 @@
+from proud.parser_wrap import parse
+from proud.basic_impl import Modular, CompilerCtx
+from proud import types
+from hybridts import type_encoding as te
+mod = parse("""
+module MyMod
+let test : exist a. a = coerce 0 : string 
+""")
+comp_ctx = CompilerCtx.top('a.prd', 'a')
+sym = comp_ctx.scope.enter("bigint")
+comp_ctx.tenv[sym] = te.App(types.type_type, types.bigint_t)
+sym = comp_ctx.scope.enter("string")
+comp_ctx.tenv[sym] = te.App(types.type_type, types.string_t)
+
+modular = Modular(comp_ctx)
+
+xs = modular.eval(mod)
+tc = comp_ctx.tc_state
+
+for k, v in comp_ctx.tenv.items():
+    print(k.name, ':', tc.infer(v), '============', v)

test/row.py

@@ -8,7 +8,7 @@
 module MyMod
 let add : bigint -> bigint -> bigint = coerce 0
 let y : {c: string} = coerce 0
-let x = {a = 0, b = 3 | y } 
+let x = {a = 0, b = 3 | y}
 """)
 comp_ctx = CompilerCtx.top('a.prd', 'a')
 sym = comp_ctx.scope.enter("bigint")