From 3da21fb367b346e8409989e272f18f91c49ab37e Mon Sep 17 00:00:00 2001
From: Monica Moniot <mamoniot@protonmail.com>
Date: Wed, 13 Dec 2023 11:01:34 -0500
Subject: [PATCH] added set seed function

---
 src/random.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/random.rs b/src/random.rs
index 1ace1e3..ed5ea6c 100644
--- a/src/random.rs
+++ b/src/random.rs
@@ -94,6 +94,14 @@ impl XorshiftRandom {
         state.jump();
         ret
     }
+
+    #[cfg(test)]
+    pub fn set_test_seed(&self, seed: u64) {
+        use std::ops::DerefMut;
+
+        let mut state = GLOBAL_XORSHIFT.lock().unwrap();
+        *Lazy::force_mut(state.deref_mut()) = Xoshiro256StarStar::seed_from_u64(seed);
+    }
 }
 impl RngCore for XorshiftRandom {
     fn next_u32(&mut self) -> u32 {