def test_success(): f.calls = 0 result = rootfinding.bracket_root(f, (0, -0.1)) check_result(result, f, f_calls=f.calls, has_bracket=True, has_root=False) assert result.iterations >= 1 rootfinding.bisect(f, result.bracket, f_bracket=result.f_bracket)
def test_instant2(): bracket = (1, -1.00001) f_bracket = (f(bracket[0]), None) f.calls = 0 result = rootfinding.bisect(f, bracket, f_bracket=f_bracket, maxiter=0) assert f.calls == 1 check_result(result, f, f_calls=f.calls, has_bracket=True, has_root=True) assert result.root == 1
def test_growth_factor2(): growth_factor = 3.14 interval = (0, -0.1) f.calls = 0 result = rootfinding.bracket_root(f, interval, growth_factor=growth_factor) check_result(result, f, f_calls=f.calls, has_bracket=True, has_root=False) assert result.bracket[1] - result.bracket[0] == pytest.approx( (interval[1] - interval[0]) * growth_factor**result.iterations)
def test_instantbracket(): interval = (0, 2) f_interval = (None, f(interval[1])) f.calls = 0 result = rootfinding.bracket_root(f, interval, f_interval=f_interval, maxiter=0) assert f.calls == 1 check_result(result, f, f_calls=f.calls, has_bracket=True, has_root=False) rootfinding.bisect(f, result.bracket, f_bracket=result.f_bracket)
def test_success(): ftol = 4 * sys.float_info.epsilon f.calls = 0 result = rootfinding.bisect(f, (0.1, -1.5), ftol=ftol) check_result(result, f, ftol=ftol, f_calls=f.calls, has_bracket=True, has_root=True) assert result.root == pytest.approx(-1)
def test_instantroot2(): interval = (-1, 0) f_interval = (f(interval[0]), None) ftol = 0 f.calls = 0 result = rootfinding.bracket_root(f, interval, f_interval=f_interval, ftol=ftol, maxiter=0) assert f.calls == 1 check_result(result, f, ftol=ftol, f_calls=f.calls, has_bracket=False, has_root=True)
def test_instantrootwithbracket(): interval = (0, -1.00001) f_interval = (f(interval[0]), None) ftol = 1e-3 f.calls = 0 result = rootfinding.bracket_root(f, interval, f_interval=f_interval, ftol=ftol, maxiter=0) assert f.calls == 1 check_result(result, f, ftol=ftol, f_calls=f.calls, has_bracket=True, has_root=True) rootfinding.bisect(f, result.bracket)