diff --git a/crates/hyperdrive-math/src/short/max.rs b/crates/hyperdrive-math/src/short/max.rs index 9ab25a9a..aa2c72bd 100644 --- 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, checkpoint_exposure: I256, diff --git a/crates/hyperdrive-math/src/short/targeted.rs b/crates/hyperdrive-math/src/short/targeted.rs index 31b51f04..ff5b8c2c 100644 --- 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()?