add proptest

scroll-tech · Dec 9, 2024 · 3c38d5d · 3c38d5d
1 parent c4a6f09
commit 3c38d5d
Show file tree

Hide file tree

Showing 3 changed files with 98 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/ceno_zkvm/Cargo.toml b/ceno_zkvm/Cargo.toml
@@ -42,6 +42,7 @@ generic_static = "0.2"
 rand.workspace = true
 tempfile = "3.14"
 thread_local = "1.1"
+proptest = "1.5.0"
 
 [dev-dependencies]
 cfg-if.workspace = true

diff --git a/ceno_zkvm/src/instructions/riscv/slt.rs b/ceno_zkvm/src/instructions/riscv/slt.rs
@@ -139,6 +139,8 @@ mod test {
 
     use rand::Rng;
 
+    use proptest::prelude::*;
+
     use super::*;
     use crate::{
         circuit_builder::{CircuitBuilder, ConstraintSystem},
@@ -244,10 +246,9 @@ mod test {
 
     #[test]
     fn test_sltu_random() {
-        let mut rng = rand::thread_rng();
-        let a: u32 = rng.gen();
-        let b: u32 = rng.gen();
-        verify::<SltuOp>("random 1", a, b, (a < b) as u32);
-        verify::<SltuOp>("random 2", b, a, (a >= b) as u32);
+        proptest::proptest!(|(a in any::<u32>(), b in any::<u32>())| {
+            verify::<SltuOp>("random 1", a, b, (a < b) as u32);
+            verify::<SltuOp>("random 2", b, a, (a >= b) as u32);
+        });
     }
 }