diff --git a/ipa-core/src/protocol/context/dzkp_validator.rs b/ipa-core/src/protocol/context/dzkp_validator.rs index 31687808b..34a8e92e8 100644 --- a/ipa-core/src/protocol/context/dzkp_validator.rs +++ b/ipa-core/src/protocol/context/dzkp_validator.rs @@ -1376,7 +1376,10 @@ mod tests { let a: Vec = repeat_with(|| rng.gen()).take(count).collect(); let b: Vec = repeat_with(|| rng.gen()).take(count).collect(); - let [ab0, ab1, ab2]: [Vec>; 3] = TestWorld::default() + let config = TestWorldConfig::default().with_timeout_secs(60); + let world = TestWorld::::with_config(&config); + + let [ab0, ab1, ab2]: [Vec>; 3] = world .malicious( zip(bit.clone(), zip(a.clone(), b.clone())), |ctx, inputs| async move { diff --git a/ipa-core/src/protocol/ipa_prf/boolean_ops/share_conversion_aby.rs b/ipa-core/src/protocol/ipa_prf/boolean_ops/share_conversion_aby.rs index ba10d84bd..74955a9e1 100644 --- a/ipa-core/src/protocol/ipa_prf/boolean_ops/share_conversion_aby.rs +++ b/ipa-core/src/protocol/ipa_prf/boolean_ops/share_conversion_aby.rs @@ -386,8 +386,9 @@ mod tests { rand::thread_rng, secret_sharing::SharedValue, seq_join::{seq_join, SeqJoin}, + sharding::NotSharded, test_executor::run, - test_fixture::{ReconstructArr, Runner, TestWorld}, + test_fixture::{ReconstructArr, Runner, TestWorld, TestWorldConfig}, }; #[test] @@ -457,7 +458,8 @@ mod tests { const COUNT: usize = CONV_CHUNK * PROOF_CHUNK * 2 + 1; const TOTAL_RECORDS: usize = COUNT.div_ceil(CONV_CHUNK); - let world = TestWorld::default(); + let config = TestWorldConfig::default().with_timeout_secs(60); + let world = TestWorld::::with_config(&config); let mut rng = thread_rng();