diff --git a/tests/unitary/pool/stateful/test_multiprecision.py b/tests/unitary/pool/stateful/test_multiprecision.py index 9fde91c4..4a7b6b05 100644 --- a/tests/unitary/pool/stateful/test_multiprecision.py +++ b/tests/unitary/pool/stateful/test_multiprecision.py @@ -1,5 +1,6 @@ import pytest from boa.test import strategy +from hypothesis import HealthCheck, settings from hypothesis.stateful import rule, run_state_machine_as_test from tests.unitary.pool.stateful.test_stateful import NumbaGoUp @@ -38,9 +39,6 @@ def exchange(self, exchange_amount_in, exchange_i, user): def test_multiprecision(users, coins, swap): - from hypothesis import settings - from hypothesis._settings import HealthCheck - Multiprecision.TestCase.settings = settings( max_examples=MAX_SAMPLES, stateful_step_count=STEP_COUNT, diff --git a/tests/unitary/pool/stateful/test_simulate.py b/tests/unitary/pool/stateful/test_simulate.py index 6da8156a..1531aba6 100644 --- a/tests/unitary/pool/stateful/test_simulate.py +++ b/tests/unitary/pool/stateful/test_simulate.py @@ -1,5 +1,6 @@ import boa from boa.test import strategy +from hypothesis import HealthCheck, settings from hypothesis.stateful import invariant, rule, run_state_machine_as_test from tests.unitary.pool.stateful.stateful_base import StatefulBase @@ -102,9 +103,6 @@ def simulator(self): def test_sim(users, coins, swap): - from hypothesis import settings - from hypothesis._settings import HealthCheck - StatefulSimulation.TestCase.settings = settings( max_examples=MAX_SAMPLES, stateful_step_count=STEP_COUNT,