test feature flag

crates/proof-of-sql/benches/bench_append_rows.rs

@@ -6,6 +6,7 @@
 //! cargo bench --features "test" --bench bench_append_rows
 //! ```
 #![allow(missing_docs, clippy::missing_docs_in_private_items)]
+use ark_std::test_rng;
 use criterion::{black_box, criterion_group, criterion_main, Criterion};
 use proof_of_sql::{
     base::{
@@ -20,7 +21,7 @@ use proof_of_sql::{
         scalar::Scalar,
     },
     proof_primitive::dory::{
-        test_rng, DoryCommitment, DoryProverPublicSetup, DoryScalar, ProverSetup, PublicParameters,
+        DoryCommitment, DoryProverPublicSetup, DoryScalar, ProverSetup, PublicParameters,
     },
 };
 use proof_of_sql_parser::posql_time::{PoSQLTimeUnit, PoSQLTimeZone};

crates/proof-of-sql/benches/jaeger_benches.rs

@@ -7,6 +7,8 @@
 //! ```
 //! Then, navigate to http://localhost:16686 to view the traces.
 
+#[cfg(feature = "test")]
+use ark_std::test_rng;
 use blitzar::{compute::init_backend, proof::InnerProductProof};
 #[cfg(feature = "test")]
 use proof_of_sql::proof_primitive::dory::{
@@ -53,10 +55,7 @@ fn main() {
         #[cfg(feature = "test")]
         "Dory" => {
             // Run 3 times to ensure that warm-up of the GPU has occurred.
-            let pp = PublicParameters::test_rand(
-                10,
-                &mut proof_of_sql::proof_primitive::dory::test_rng(),
-            );
+            let pp = PublicParameters::test_rand(10, &mut test_rng());
             let ps = ProverSetup::from(&pp);
             let prover_setup = DoryProverPublicSetup::new(&ps, 10);
             let vs = VerifierSetup::from(&pp);

crates/proof-of-sql/src/proof_primitive/dory/mod.rs

@@ -27,7 +27,7 @@ mod rand_util;
 use rand_util::rand_F_tensors;
 #[cfg(test)]
 use rand_util::rand_G_vecs;
-#[cfg(any(test, feature = "test"))]
+#[cfg(test)]
 pub use rand_util::test_rng;
 
 mod dory_messages;

crates/proof-of-sql/src/proof_primitive/dory/rand_util.rs

@@ -1,10 +1,12 @@
 #[cfg(test)]
-use super::F;
-use super::{G1Affine, G2Affine};
+use super::{G1Affine, G2Affine, F};
 #[cfg(test)]
-use ark_std::rand::{rngs::StdRng, SeedableRng};
-use ark_std::{rand::Rng, UniformRand};
+use ark_std::{
+    rand::{rngs::StdRng, Rng, SeedableRng},
+    UniformRand,
+};
 
+#[cfg(test)]
 /// Create a random number generator for testing.
 pub fn test_rng() -> impl Rng {
     ark_std::test_rng()
@@ -16,7 +18,7 @@ pub fn test_seed_rng(seed: [u8; 32]) -> impl Rng {
     StdRng::from_seed(seed)
 }
 
-#[allow(dead_code)]
+#[cfg(test)]
 /// Creates two vectors of random G1 and G2 elements with length 2^nu.
 pub fn rand_G_vecs<R>(nu: usize, rng: &mut R) -> (Vec<G1Affine>, Vec<G2Affine>)
 where

crates/proof-of-sql/tests/timestamp_integration_tests.rs

@@ -1,11 +1,12 @@
 #![cfg(feature = "test")]
+use ark_std::test_rng;
 #[cfg(feature = "blitzar")]
 use proof_of_sql::base::commitment::InnerProductProof;
 use proof_of_sql::{
     base::database::{owned_table_utility::*, OwnedTableTestAccessor, TestAccessor},
     proof_primitive::dory::{
-        test_rng, DoryCommitment, DoryEvaluationProof, DoryProverPublicSetup,
-        DoryVerifierPublicSetup, ProverSetup, PublicParameters, VerifierSetup,
+        DoryCommitment, DoryEvaluationProof, DoryProverPublicSetup, DoryVerifierPublicSetup,
+        ProverSetup, PublicParameters, VerifierSetup,
     },
     sql::{
         parse::QueryExpr,