from __future__ import absolute_import, print_function
import numpy as np
from numpy.testing import assert_array_almost_equal, assert_equal
import numdifftools.core as nd
from numdifftools.step_generators import EPS
[docs]
def test_min_step_generator_with_step_ratio4():
step_gen = nd.MinStepGenerator(base_step=None, num_steps=10, step_ratio=4, offset=-1)
h = np.array(list(step_gen(0)))
desired = np.array(
[
3.58968236e-02,
8.97420590e-03,
2.24355147e-03,
5.60887869e-04,
1.40221967e-04,
3.50554918e-05,
8.76387295e-06,
2.19096824e-06,
5.47742059e-07,
1.36935515e-07,
]
)
assert_array_almost_equal((h - desired) / desired, 0)
[docs]
def test_min_step_generator_default_base_step():
step_gen = nd.MinStepGenerator(num_steps=1, offset=0)
h = list(step_gen(0))
desired = EPS ** (1.0 / 2.5)
assert_array_almost_equal((h[-1] - desired) / desired, 0)
[docs]
def test__min_step_generator_with_step_nom1():
step_gen = nd.MinStepGenerator(num_steps=1, step_nom=1.0, offset=0)
h = list(step_gen(0))
desired = EPS ** (1.0 / 2.5)
assert_array_almost_equal((h[-1] - desired) / desired, 0)
[docs]
def test_min_step_generator_with_base_step01():
desired = 0.1
step_gen = nd.MinStepGenerator(base_step=desired, num_steps=1, offset=0)
methods = ["forward", "backward", "central", "complex"]
for n in range(1, 5):
for order in [1, 2, 4, 6, 8]:
min_length = n + order - 1
lengths = [min_length, min_length, max(min_length // 2, 1), max(min_length // 4, 1)]
for m, method in zip(lengths, methods, strict=False):
h = list(step_gen(0, method=method, n=n, order=order))
# print(len(h), n, order, method)
assert_array_almost_equal((h[-1] - desired) / desired, 0)
assert_equal(m, len(h))
[docs]
def test_default_max_step_generator():
step_gen = nd.MaxStepGenerator(num_steps=10)
h = np.array(list(step_gen(0)))
desired = 2.0 * 2.0 ** (-np.arange(10) + 0)
assert_array_almost_equal((h - desired) / desired, 0)
[docs]
def test_max_step_generator_default_base_step():
step_gen = nd.MaxStepGenerator(num_steps=1, offset=0)
h = list(step_gen(0))
desired = 2.0
assert_array_almost_equal((h[0] - desired) / desired, 0)
[docs]
def test_max_step_generator_with_base_step01():
desired = 0.1
step_gen = nd.MaxStepGenerator(base_step=desired, num_steps=1, offset=0)
methods = ["forward", "backward", "central", "complex"]
lengths = [2, 2, 1, 1]
for n, method in zip(lengths, methods, strict=False):
h = list(step_gen(0, method=method))
assert_array_almost_equal((h[0] - desired) / desired, 0)
assert_equal(n, len(h))