another test fix

aiken-lang · Jan 23, 2024 · b993159 · b993159
1 parent 8249fca
commit b993159
Showing 1 changed file with 15 additions and 9 deletions.
diff --git a/crates/aiken-project/src/tests/gen_uplc.rs b/crates/aiken-project/src/tests/gen_uplc.rs
@@ -4303,22 +4303,28 @@ fn expect_head_no_tail() {
 
     assert_uplc(
         src,
-        Term::tail_list()
-            .apply(Term::var("a"))
+        Term::var("a")
             .delayed_choose_list(
-                Term::equals_integer()
-                    .apply(Term::var("h"))
-                    .apply(Term::var("h")),
-                Term::Error.delayed_trace(Term::string("expect [h] = a")),
+                Term::Error.delayed_trace(Term::var("expect[h]=a")),
+                Term::tail_list()
+                    .apply(Term::var("a"))
+                    .delayed_choose_list(
+                        Term::equals_integer()
+                            .apply(Term::var("h"))
+                            .apply(Term::var("h")),
+                        Term::Error.delayed_trace(Term::var("expect[h]=a")),
+                    )
+                    .lambda("h")
+                    .apply(Term::un_i_data().apply(Term::head_list().apply(Term::var("a")))),
             )
-            .lambda("h")
-            .apply(Term::un_i_data().apply(Term::head_list().apply(Term::var("a"))))
             .lambda("a")
             .apply(Term::list_values(vec![
                 Constant::Data(Data::integer(1.into())),
                 Constant::Data(Data::integer(2.into())),
                 Constant::Data(Data::integer(3.into())),
-            ])),
+            ]))
+            .lambda("expect[h]=a")
+            .apply(Term::string("expect [h] = a")),
         true,
     );
 }