From aaaee7bdc476f5f0631dc0d2f367c54bdfe03d12 Mon Sep 17 00:00:00 2001
From: Arnaud Bailly <arnaud.bailly@iohk.io>
Date: Sun, 6 Oct 2024 16:05:32 +0200
Subject: Restructure Client to contain all test execution logic

---
 rust/src/lambda.rs | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

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

diff --git a/rust/src/lambda.rs b/rust/src/lambda.rs
index 7c8d9be..d1ab5b5 100644
--- a/rust/src/lambda.rs
+++ b/rust/src/lambda.rs
@@ -1,4 +1,4 @@
-use rand::Rng;
+use rand::{rngs::SmallRng, Rng, RngCore, SeedableRng};
 use serde::{Deserialize, Serialize};
 use std::collections::HashMap;
 
@@ -98,8 +98,14 @@ fn gensym() -> String {
     format!("x_{}", n1)
 }
 
-pub fn generate_expr(size: u32) -> Value {
-    Value::Num(3)
+pub fn generate_expr<R: Rng>(size: u32, rng: &mut R) -> Value {
+    match size {
+        0 | 1 => {
+            let n: u16 = rng.gen();
+            Value::Num(n.into())
+        }
+        _ => todo!(),
+    }
 }
 
 #[cfg(test)]
-- 
cgit v1.2.3