error handling

tree_unique_args/src/type_analysis.egg

@@ -225,11 +225,28 @@
       ((HasType f (FuncT in-ty out-ty)))
       :ruleset type-analysis)
 
+(rule (
+       (= f (Function id body in-ty out-ty))
+       (HasType body ty)
+       (!= ty out-ty)
+      )
+      ((panic "Function body had wrong type :("))
+      :ruleset type-analysis)
+
 (rule (
        (= c (Call f-id arg))
        (= f (Function f-id body in-ty out-ty))
        (HasType f (FuncT in-ty out-ty))
        (HasType arg in-ty)
       )
       ((HasType c out-ty))
-      :ruleset type-analysis)
+      :ruleset type-analysis)
+
+(rule (
+       (= c (Call f-id arg))
+       (=  f (Function f-id body in-ty out-ty))
+       (HasType arg ty)
+       (!= ty in-ty)
+      )
+      ((panic "Function call had wrong argument type :("))
+      :ruleset type-analysis)

tree_unique_args/src/type_analysis.rs

@@ -255,3 +255,38 @@ fn func() -> Result<(), egglog::Error> {
     );
     crate::run_test(build, &check)
 }
+
+#[test]
+#[should_panic]
+fn func_input_type() {
+    let build = "
+    (let ctx (Id 0))
+    (let f-id (Id 1))
+    (let f (Function f-id (Add (Arg f-id) (Num f-id 2)) (IntT) (IntT)))
+    (let c (Call f-id (Boolean ctx true)))
+  ";
+    let check = format!(
+        "
+    {SCHED}
+    "
+    );
+
+    let _ = crate::run_test(build, &check);
+}
+
+#[test]
+#[should_panic]
+fn func_output_type() {
+    let build = "
+    (let ctx (Id 0))
+    (let f-id (Id 1))
+    (let f (Function f-id (Add (Arg f-id) (Num f-id 2)) (IntT) (BoolT)))
+  ";
+    let check = format!(
+        "
+    {SCHED}
+    "
+    );
+
+    let _ = crate::run_test(build, &check);
+}