test: add linear relation with subtraction.

Author: mmaker

src/tests/test_relations.rs

@@ -5,6 +5,7 @@ use group::prime::PrimeGroup;
 use rand::rngs::OsRng;
 use rand::RngCore;
 
+use crate::composition::ComposedRelation;
 use crate::fiat_shamir::Nizk;
 use crate::{
     codec::Shake128DuplexSponge, linear_relation::CanonicalLinearRelation,
@@ -266,6 +267,28 @@ pub fn weird_linear_combination<G: PrimeGroup, R: RngCore>(
     (instance, witness)
 }
 
+
+fn test_relation_with_subtractions<G: PrimeGroup, R: RngCore>(
+    mut rng: &mut R,
+) -> (CanonicalLinearRelation<G>, Vec<G::Scalar>) {
+    use ff::PrimeField;
+
+    let x = G::Scalar::random(&mut rng);
+    let B = G::random(&mut rng);
+    let X = B * (x - G::Scalar::from_u128(1u128));
+
+    let mut linear_relation = LinearRelation::<G>::new();
+    let x_var = linear_relation.allocate_scalar();
+    let B_var = linear_relation.allocate_element();
+    let X_var = linear_relation
+        .allocate_eq((x_var + (-G::Scalar::from_u128(1u128))) * B_var);
+    linear_relation.set_element(B_var, B);
+    linear_relation.set_element(X_var, X);
+
+    let instance = (&linear_relation).try_into().unwrap();
+    let witness = vec![x];
+    (instance, witness)
+}
 /// Generic helper function to test both relation correctness and NIZK functionality
 #[test]
 fn test_common_relations() {
@@ -291,6 +314,7 @@ fn test_common_relations() {
         "weird_linear_combination",
         Box::new(weird_linear_combination),
     );
+    instance_generators.insert("test_relation_with_subtractions", Box::new(test_relation_with_subtractions));
 
     for (relation_name, relation_sampler) in instance_generators.iter() {
         let mut rng = OsRng;