From 1ed2e1a8e98a31f4fab49b70fbf23cbbd8e3c024 Mon Sep 17 00:00:00 2001
From: ayazhafiz <ayaz.hafiz.1@gmail.com>
Date: Tue, 28 Jun 2022 22:16:48 -0400
Subject: [PATCH] Improve test

---
 .../compiler/test_gen/src/gen_primitives.rs   | 21 +++++++++----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/crates/compiler/test_gen/src/gen_primitives.rs b/crates/compiler/test_gen/src/gen_primitives.rs
index 9be5b1ca43..458990b60c 100644
--- a/crates/compiler/test_gen/src/gen_primitives.rs
+++ b/crates/compiler/test_gen/src/gen_primitives.rs
@@ -3607,21 +3607,20 @@ fn multimorphic_lambdas_have_captured_function_in_closure() {
                         b -> b {}
                 thunk
 
-            f = \_ -> \_ -> ""
+            f = \_ -> \_ -> "fun f"
             g = \{ s1 } -> \_ -> s1
 
-            x : [True, False]
-            x = False
+            fun = \x ->
+                h = 
+                    when x is
+                        True -> after (\{} -> "") f
+                        False -> after (\{} -> {s1: "s1"}) g
+                h {}
 
-            fun =
-                when x is
-                    True -> after (\{} -> "") f
-                    False -> after (\{} -> {s1: "s1"}) g
-
-            fun {}
+            {a: fun False, b: fun True}
             "#
         ),
-        RocStr::from("s1"),
-        RocStr
+        (RocStr::from("s1"), RocStr::from("fun f")),
+        (RocStr, RocStr)
     )
 }