cleanup

delvtech · Aug 13, 2024 · 70e9ddf · 70e9ddf
1 parent 8167307
commit 70e9ddf
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/crates/hyperdrive-math/src/short/max.rs b/crates/hyperdrive-math/src/short/max.rs
@@ -480,7 +480,7 @@ impl State {
     /// ```math
     /// e(x) = e_0 - max(e_{c}, 0)
     /// ```
-    fn solvency_after_short(
+    pub fn solvency_after_short(
         &self,
         bond_amount: FixedPoint<U256>,
         checkpoint_exposure: I256,

diff --git a/crates/hyperdrive-math/src/short/targeted.rs b/crates/hyperdrive-math/src/short/targeted.rs
@@ -382,10 +382,17 @@ impl State {
 
 #[cfg(test)]
 mod tests {
-    use hyperdrive_test_utils::constants::FAST_FUZZ_RUNS;
-    use rand::{thread_rng, Rng};
+    use hyperdrive_test_utils::{
+        chain::TestChain,
+        constants::{FAST_FUZZ_RUNS, SLOW_FUZZ_RUNS},
+    };
+    use rand::{thread_rng, Rng, SeedableRng};
+    use rand_chacha::ChaCha8Rng;
 
     use super::*;
+    use crate::test_utils::{
+        agent::HyperdriveMathAgent, preamble::initialize_pool_with_random_state,
+    };
 
     #[tokio::test]
     async fn fuzz_short_trade_deltas_from_reserves() -> Result<()> {
@@ -481,7 +488,7 @@ mod tests {
             // Run the preamble and get state information.
             initialize_pool_with_random_state(&mut rng, &mut alice, &mut bob, &mut celine).await?;
             let current_state = alice.get_state().await?;
-            let min_spot_price_before_short = current_state.calculate_min_price()?;
+            let min_spot_price_before_short = current_state.calculate_min_spot_price()?;
 
             // Get a targeted short amount.
             let target_rate = current_state.calculate_spot_rate()?