wip

Author: ratankaliani

plonky2x/core/src/frontend/ops/math.rs

@@ -172,11 +172,9 @@ impl<L: PlonkParameters<D>, const D: usize> CircuitBuilder<L, D> {
     pub fn lt<Lhs, Rhs>(&mut self, lhs: Lhs, rhs: Rhs) -> BoolVariable
     where
         Lhs: LessThanOrEqual<L, D, Lhs>,
-        Rhs: Sub<L, D, Rhs, Output = Lhs> + One<L, D>,
     {
-        let one = self.one::<Rhs>();
-        let upper_bound = rhs.sub(one, self);
-        self.lte(lhs, upper_bound)
+        let lte = rhs.lte(lhs, self);
+        self.not(lte)
     }
 
     /// The greater than operation (>).
@@ -226,6 +224,64 @@ mod tests {
     #[allow(unused_imports)]
     use crate::prelude::{BoolVariable, DefaultBuilder, U32Variable};
 
+    #[test]
+    fn test_math_lt() {
+        let mut builder = DefaultBuilder::new();
+
+        let v0 = builder.read::<U32Variable>();
+        let v1 = builder.read::<U32Variable>();
+        let result = builder.read::<BoolVariable>();
+        let computed_result = builder.lt(v0, v1);
+        builder.assert_is_equal(result, computed_result);
+
+        let circuit = builder.build();
+
+        let test_cases = [
+            (5u32, 0u32, false),
+            (0u32, 10u32, true),
+            (10u32, 5u32, false),
+        ];
+
+        for test_case in test_cases.iter() {
+            let mut input = circuit.input();
+            input.write::<U32Variable>(test_case.0);
+            input.write::<U32Variable>(test_case.1);
+            input.write::<BoolVariable>(test_case.2);
+
+            let (proof, output) = circuit.prove(&input);
+            circuit.verify(&proof, &input, &output);
+        }
+    }
+
+    #[test]
+    fn test_math_lte() {
+        let mut builder = DefaultBuilder::new();
+
+        let v0 = builder.read::<U32Variable>();
+        let v1 = builder.read::<U32Variable>();
+        let result = builder.read::<BoolVariable>();
+        let computed_result = builder.lte(v0, v1);
+        builder.assert_is_equal(result, computed_result);
+
+        let circuit = builder.build();
+
+        let test_cases = [
+            (0u32, 0u32, true),
+            (0u32, 100u32, true),
+            (10u32, 0u32, false),
+        ];
+
+        for test_case in test_cases.iter() {
+            let mut input = circuit.input();
+            input.write::<U32Variable>(test_case.0);
+            input.write::<U32Variable>(test_case.1);
+            input.write::<BoolVariable>(test_case.2);
+
+            let (proof, output) = circuit.prove(&input);
+            circuit.verify(&proof, &input, &output);
+        }
+    }
+
     #[test]
     fn test_math_gt() {
         let mut builder = DefaultBuilder::new();