From bf16d83e52f04a644aba84c8136f049ed4f87380 Mon Sep 17 00:00:00 2001 From: Chan-Ho Suh Date: Mon, 7 Aug 2023 19:07:53 -0400 Subject: [PATCH] Fix broken import in test module --- test/unit/test_param_samplers.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/test/unit/test_param_samplers.py b/test/unit/test_param_samplers.py index 95927f1bd..6f3bb1b7f 100644 --- a/test/unit/test_param_samplers.py +++ b/test/unit/test_param_samplers.py @@ -1,8 +1,7 @@ -import pytest - from math import prod -from hypothesis import given, settings, HealthCheck +import pytest +from hypothesis import HealthCheck, given, settings from hypothesis import strategies as st from curvesim.exceptions import ParameterSamplerError @@ -10,10 +9,9 @@ from curvesim.iterators.param_samplers.parameterized_pool_iterator import ( DEFAULT_POOL_MAP, ) -from curvesim.pool.cryptoswap.pool import _newton_D +from curvesim.pool.cryptoswap.calcs import newton_D from curvesim.pool.sim_interface import SimCurveCryptoPool - # Strategies POOL_PARAMS = { "A": st.integers(min_value=1, max_value=3000), @@ -315,7 +313,7 @@ def _test_pool_params(pool, params): if _key == "D": if isinstance(pool, SimCurveCryptoPool): - D_from_xp = _newton_D(_pool.A, _pool.gamma, _pool._xp()) + D_from_xp = newton_D(_pool.A, _pool.gamma, _pool._xp()) assert _pool.D == val assert ( abs(D_from_xp - val) <= 10**10