Add impl Distribution<MerkleHashOrchard> for Standard for testing.

zcash · Mar 16, 2024 · fca9ec6 · fca9ec6
1 parent e74879d
commit fca9ec6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/tree.rs b/src/tree.rs
@@ -264,14 +264,20 @@ impl<'de> Deserialize<'de> for MerkleHashOrchard {
 #[cfg(feature = "test-dependencies")]
 pub mod testing {
     use ff::Field;
-    use rand::RngCore;
+    use rand::{RngCore, distributions::{Standard, Distribution}};
 
     use super::MerkleHashOrchard;
 
     impl MerkleHashOrchard {
         /// Return a random fake `MerkleHashOrchard`.
         pub fn random(rng: &mut impl RngCore) -> Self {
-            Self(pasta_curves::Fp::random(rng))
+            Standard.sample(rng)
+        }
+    }
+
+    impl Distribution<MerkleHashOrchard> for Standard {
+        fn sample<R: rand::Rng + ?Sized>(&self, rng: &mut R) -> MerkleHashOrchard {
+            MerkleHashOrchard(pasta_curves::Fp::random(rng))
         }
     }
 }