From 3e03398d9b5e1212dd0686893fa3a5950fa3c9b5 Mon Sep 17 00:00:00 2001
From: pca006132 <john.lck40@gmail.com>
Date: Wed, 21 Jul 2021 15:59:01 +0800
Subject: [PATCH] obj test

---
 nac3core/src/typecheck/type_inferencer/mod.rs |  2 +-
 .../src/typecheck/type_inferencer/test.rs     | 50 ++++++++++++++-----
 2 files changed, 39 insertions(+), 13 deletions(-)

diff --git a/nac3core/src/typecheck/type_inferencer/mod.rs b/nac3core/src/typecheck/type_inferencer/mod.rs
index 7cd72b41d..23b9660e3 100644
--- a/nac3core/src/typecheck/type_inferencer/mod.rs
+++ b/nac3core/src/typecheck/type_inferencer/mod.rs
@@ -105,7 +105,7 @@ impl<'a> fold::Fold<()> for Inferencer<'a> {
                         .unify(target.custom.unwrap(), value.custom.unwrap())?;
                 }
             }
-            ast::StmtKind::AnnAssign { .. } => {}
+            ast::StmtKind::AnnAssign { .. } | ast::StmtKind::Expr { .. } => {}
             _ => return Err("Unsupported statement type".to_string()),
         };
         Ok(stmt)
diff --git a/nac3core/src/typecheck/type_inferencer/test.rs b/nac3core/src/typecheck/type_inferencer/test.rs
index 23f996449..c06208a3e 100644
--- a/nac3core/src/typecheck/type_inferencer/test.rs
+++ b/nac3core/src/typecheck/type_inferencer/test.rs
@@ -66,11 +66,7 @@ impl TestEnvironment {
             fields: HashMap::new(),
             params: HashMap::new(),
         });
-        type_mapping.insert("int32".into(), int32);
-        type_mapping.insert("int64".into(), int64);
-        type_mapping.insert("float".into(), float);
-        type_mapping.insert("bool".into(), bool);
-        type_mapping.insert("none".into(), none);
+        type_mapping.insert("None".into(), none);
 
         let primitives = PrimitiveStore {
             int32,
@@ -81,13 +77,20 @@ impl TestEnvironment {
         };
 
         let (v0, id) = unifier.get_fresh_var();
+
+        let foo_ty = unifier.add_ty(TypeEnum::TObj {
+            obj_id: 5,
+            fields: [("a".into(), v0)].iter().cloned().collect(),
+            params: [(id, v0)].iter().cloned().collect(),
+        });
+
         type_mapping.insert(
-            "foo".into(),
-            unifier.add_ty(TypeEnum::TObj {
-                obj_id: 5,
-                fields: [("a".into(), v0)].iter().cloned().collect(),
-                params: [(id, v0)].iter().cloned().collect(),
-            }),
+            "Foo".into(),
+            unifier.add_ty(TypeEnum::TFunc(FunSignature {
+                args: vec![],
+                ret: foo_ty,
+                vars: [(id, v0)].iter().cloned().collect(),
+            })),
         );
 
         let id_to_name = [
@@ -140,6 +143,22 @@ impl TestEnvironment {
     "},
     [("a", "fn[[x=float, y=float], float]"), ("b", "fn[[x=float], float]"), ("c", "float"), ("d", "float")].iter().cloned().collect()
     ; "lambda test")]
+#[test_case(indoc! {"
+        a = lambda x: x
+        b = lambda x: x
+
+        foo1 = Foo()
+        foo2 = Foo()
+        c = a(foo1.a)
+        d = b(foo2.a)
+
+        a(True)
+        b(123)
+
+    "},
+    [("a", "fn[[x=bool], bool]"), ("b", "fn[[x=int32], int32]"), ("c", "bool"),
+     ("d", "int32"), ("foo1", "Foo[bool]"), ("foo2", "Foo[int32]")].iter().cloned().collect()
+    ; "obj test")]
 fn test_basic(source: &str, mapping: HashMap<&str, &str>) {
     let mut env = TestEnvironment::new();
     let id_to_name = std::mem::take(&mut env.id_to_name);
@@ -150,6 +169,14 @@ fn test_basic(source: &str, mapping: HashMap<&str, &str>) {
         .map(|v| inferencer.fold_stmt(v))
         .collect::<Result<Vec<_>, _>>()
         .unwrap();
+    for (k, v) in inferencer.variable_mapping.iter() {
+        let name = inferencer.unifier.stringify(
+            *v,
+            &mut |v| id_to_name.get(&v).unwrap().clone(),
+            &mut |v| format!("v{}", v),
+        );
+        println!("{}: {}", k, name);
+    }
     for (k, v) in mapping.iter() {
         let ty = inferencer.variable_mapping.get(*k).unwrap();
         let name = inferencer.unifier.stringify(
@@ -160,4 +187,3 @@ fn test_basic(source: &str, mapping: HashMap<&str, &str>) {
         assert_eq!(format!("{}: {}", k, v), format!("{}: {}", k, name));
     }
 }
-