From d3781941c182fdc2971bac09b33b273e46232b00 Mon Sep 17 00:00:00 2001
From: Arnaud Bailly <arnaud.bailly@iohk.io>
Date: Wed, 25 Sep 2024 08:15:21 +0200
Subject: Evaluate both side of application

---
 rust/src/parser.rs | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'rust/src/parser.rs')

diff --git a/rust/src/parser.rs b/rust/src/parser.rs
index c85a425..9dc0bb8 100644
--- a/rust/src/parser.rs
+++ b/rust/src/parser.rs
@@ -250,12 +250,19 @@ mod tests {
         type Strategy = BoxedStrategy<Self>;
 
         fn arbitrary_with(_args: ()) -> Self::Strategy {
-            prop_oneof![
+            let identifier = "\\pL(\\pL|\\pN)*";
+            let leaf = prop_oneof![
                 any::<i32>().prop_map(Num),
                 any::<bool>().prop_map(Bool),
                 // see https://unicode.org/reports/tr18/#General_Category_Property for one letter unicode categories
-                "\\pL(\\pL|\\pN)*".prop_map(Sym),
-            ]
+                identifier.prop_map(Sym),
+            ];
+            leaf.prop_recursive(4, 128, 5, move |inner| {
+                prop_oneof![
+                    (inner.clone(), inner.clone()).prop_map(|(l, r)| App(Box::new(l), Box::new(r))),
+                    (identifier, inner).prop_map(|(var, body)| Lam(var, Box::new(body))),
+                ]
+            })
             .boxed()
         }
     }
-- 
cgit v1.2.3