add new test

aiken-lang · Jan 14, 2024 · 06672fc · 06672fc
1 parent d265240
commit 06672fc
Showing 1 changed file with 50 additions and 0 deletions.
diff --git a/crates/aiken-project/src/tests/gen_uplc.rs b/crates/aiken-project/src/tests/gen_uplc.rs
@@ -5557,6 +5557,56 @@ fn expect_none() {
     );
 }
 
+#[test]
+fn head_list_on_map() {
+    let src = r#"
+        use aiken/builtin
+
+        test exp_none() {
+          let x = [(1, ""), (2, #"aa")]
+          builtin.head_list(x) == (1, "")
+        }
+  "#;
+
+    assert_uplc(
+        src,
+        Term::equals_data()
+            .apply(
+                Term::map_data().apply(
+                    Term::mk_cons()
+                        .apply(Term::head_list().apply(Term::var("x")))
+                        .apply(Term::empty_map()),
+                ),
+            )
+            .apply(
+                Term::map_data().apply(
+                    Term::mk_cons()
+                        .apply(Term::pair_values(
+                            Constant::Data(Data::integer(1.into())),
+                            Constant::Data(Data::bytestring(vec![])),
+                        ))
+                        .apply(Term::empty_map()),
+                ),
+            )
+            .lambda("x")
+            .apply(Term::map_values(vec![
+                Constant::ProtoPair(
+                    Type::Data,
+                    Type::Data,
+                    Constant::Data(Data::integer(1.into())).into(),
+                    Constant::Data(Data::bytestring(vec![])).into(),
+                ),
+                Constant::ProtoPair(
+                    Type::Data,
+                    Type::Data,
+                    Constant::Data(Data::integer(2.into())).into(),
+                    Constant::Data(Data::bytestring(vec![170])).into(),
+                ),
+            ])),
+        false,
+    );
+}
+
 #[test]
 fn tuple_2_match() {
     let src = r#"