Fix tests when odd-primes is disabled

JoeyBF · Aug 14, 2024 · c89c833 · c89c833
1 parent 6764278
commit c89c833
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/ext/crates/fp/src/field/fp.rs b/ext/crates/fp/src/field/fp.rs
@@ -190,7 +190,7 @@ mod tests {
 
         fn arb_elements<const N: usize>(
         ) -> impl Strategy<Value = (Fp<ValidPrime>, [FieldElement<Fp<ValidPrime>>; N])> {
-            crate::prime::tests::arb_prime().prop_flat_map(|p| super::arb_elements(p))
+            crate::prime::tests::arb_prime().prop_flat_map(super::arb_elements)
         }
 
         field_tests!();
@@ -218,7 +218,9 @@ mod tests {
     }
 
     static_fp_tests!(P2);
-    static_fp_tests!(P3);
-    static_fp_tests!(P5);
-    static_fp_tests!(P7);
+    cfg_if::cfg_if! { if #[cfg(feature = "odd-primes")] {
+        static_fp_tests!(P3);
+        static_fp_tests!(P5);
+        static_fp_tests!(P7);
+    }}
 }
diff --git a/ext/crates/fp/src/field/smallfq.rs b/ext/crates/fp/src/field/smallfq.rs
@@ -351,7 +351,7 @@ mod tests {
         fn arb_elements<const N: usize>(
         ) -> impl Strategy<Value = (SmallFq<ValidPrime>, [FieldElement<SmallFq<ValidPrime>>; N])>
         {
-            arb_smallfq_prime().prop_flat_map(|p| super::arb_elements(p))
+            arb_smallfq_prime().prop_flat_map(super::arb_elements)
         }
 
         field_tests!();
@@ -380,7 +380,9 @@ mod tests {
     }
 
     static_smallfq_tests!(P2);
-    static_smallfq_tests!(P3);
-    static_smallfq_tests!(P5);
-    static_smallfq_tests!(P7);
+    cfg_if::cfg_if! { if #[cfg(feature = "odd-primes")] {
+        static_smallfq_tests!(P3);
+        static_smallfq_tests!(P5);
+        static_smallfq_tests!(P7);
+    }}
 }