macro_rules! test_uint { ($name: ident, $_type: ty, $integer_type: expr, $gadget: ty) => { pub struct $name {} impl $name { fn test_min(min: $_type) { let min_allocated = <$gadget>::constant(min); let bytes = include_bytes!("min.leo"); let program = parse_program(bytes).unwrap(); output_expected_allocated(program, min_allocated); } fn test_max(max: $_type) { let max_allocated = <$gadget>::constant(max); let bytes = include_bytes!("max.leo"); let program = parse_program(bytes).unwrap(); output_expected_allocated(program, max_allocated); } } impl IntegerTester for $name { fn test_input() { // valid input let num: $_type = rand::random(); let expected = <$gadget>::constant(num); let bytes = include_bytes!("input.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![Some(InputValue::Integer($integer_type, num as u128))]); output_expected_allocated(program, expected); // invalid input let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![Some(InputValue::Boolean(true))]); fail_integer(program); // None input let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![None]); fail_integer(program); } fn test_add() { for _ in 0..10 { let r1: $_type = rand::random(); let r2: $_type = rand::random(); let sum = r1.wrapping_add(r2); let cs = TestConstraintSystem::::new(); let sum_allocated = <$gadget>::alloc(cs, || Ok(sum)).unwrap(); let bytes = include_bytes!("add.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_allocated(program, sum_allocated); } } fn test_sub() { for _ in 0..10 { let r1: $_type = rand::random(); let r2: $_type = rand::random(); let difference = r1.wrapping_sub(r2); let cs = TestConstraintSystem::::new(); let difference_allocated = <$gadget>::alloc(cs, || Ok(difference)).unwrap(); let bytes = include_bytes!("sub.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_allocated(program, difference_allocated); } } fn test_mul() { for _ in 0..10 { let r1: $_type = rand::random(); let r2: $_type = rand::random(); let product = r1.wrapping_mul(r2); let cs = TestConstraintSystem::::new(); let product_allocated = <$gadget>::alloc(cs, || Ok(product)).unwrap(); let bytes = include_bytes!("mul.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_allocated(program, product_allocated); } } fn test_div() { for _ in 0..10 { let r1: $_type = rand::random(); let r2: $_type = rand::random(); let bytes = include_bytes!("div.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); // expect an error when dividing by zero if r2 == 0 { let _err = get_error(program); } else { let cs = TestConstraintSystem::::new(); let quotient = r1.wrapping_div(r2); let quotient_allocated = <$gadget>::alloc(cs, || Ok(quotient)).unwrap(); output_expected_allocated(program, quotient_allocated); } } } fn test_pow() { // for _ in 0..10 {// these loops take an excessive amount of time let r1: $_type = rand::random(); let r2: $_type = rand::random(); let r2 = r2 as u32; // we cast to u32 here because of rust pow() requirements let result = r1.wrapping_pow(r2); let cs = TestConstraintSystem::::new(); let result_allocated = <$gadget>::alloc(cs, || Ok(result)).unwrap(); let bytes = include_bytes!("pow.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_allocated(program, result_allocated); // } } fn test_eq() { for _ in 0..10 { let r1: $_type = rand::random(); // test equal let bytes = include_bytes!("eq.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r1 as u128)), ]); output_true(program); // test not equal let r2: $_type = rand::random(); let result = r1.eq(&r2); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_boolean(program, result); } } fn test_ge() { for _ in 0..10 { let r1: $_type = rand::random(); // test equal let bytes = include_bytes!("ge.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r1 as u128)), ]); output_true(program); // test not equal let r2: $_type = rand::random(); let result = r1.ge(&r2); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_boolean(program, result); } } fn test_gt() { for _ in 0..10 { let r1: $_type = rand::random(); // test equal let bytes = include_bytes!("gt.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r1 as u128)), ]); output_false(program); // test not equal let r2: $_type = rand::random(); let result = r1.gt(&r2); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_boolean(program, result); } } fn test_le() { for _ in 0..10 { let r1: $_type = rand::random(); // test equal let bytes = include_bytes!("le.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r1 as u128)), ]); output_true(program); // test not equal let r2: $_type = rand::random(); let result = r1.le(&r2); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_boolean(program, result); } } fn test_lt() { for _ in 0..10 { let r1: $_type = rand::random(); // test equal let bytes = include_bytes!("lt.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r1 as u128)), ]); output_false(program); // test not equal let r2: $_type = rand::random(); let result = r1.lt(&r2); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_boolean(program, result); } } fn test_assert_eq() { for _ in 0..10 { let r1: $_type = rand::random(); // test equal let bytes = include_bytes!("assert_eq.leo"); let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r1 as u128)), ]); let _ = get_output(program); // test not equal let r2: $_type = rand::random(); if r1 == r2 { continue; } let mut program = parse_program(bytes).unwrap(); program.set_inputs(vec![ Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); let mut cs = TestConstraintSystem::::new(); let _ = program.compile_constraints(&mut cs).unwrap(); assert!(!cs.is_satisfied()); } } fn test_ternary() { let r1: $_type = rand::random(); let r2: $_type = rand::random(); let g1 = <$gadget>::constant(r1); let g2 = <$gadget>::constant(r2); let bytes = include_bytes!("ternary.leo"); let mut program_1 = parse_program(bytes).unwrap(); let mut program_2 = program_1.clone(); // true -> field 1 program_1.set_inputs(vec![ Some(InputValue::Boolean(true)), Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_allocated(program_1, g1); // false -> field 2 program_2.set_inputs(vec![ Some(InputValue::Boolean(false)), Some(InputValue::Integer($integer_type, r1 as u128)), Some(InputValue::Integer($integer_type, r2 as u128)), ]); output_expected_allocated(program_2, g2); } } }; }