def test_opt_params_poly_factory(order): """Tests that optimal parameters are stored after calling the reduce method. """ fac = PolyFactory(scale_factors=np.linspace(1, 10, 10), order=order) assert fac.opt_params == [] fac.iterate(apply_seed_to_func(f_non_lin, seed=SEED)) zne_value = fac.reduce() assert len(fac.opt_params) == order + 1 assert np.isclose(fac.opt_params[-1], zne_value)
def test_poly_extr(): """Test of polynomial extrapolator.""" # test (order=1) fac = PolyFactory(X_VALS, order=1) fac.iterate(f_lin) assert np.isclose(fac.reduce(), f_lin(0, err=0), atol=CLOSE_TOL) # test that, for some non-linear functions, # order=1 is bad while order=2 is better. seeded_f = apply_seed_to_func(f_non_lin, SEED) fac = PolyFactory(X_VALS, order=1) fac.iterate(seeded_f) assert not np.isclose(fac.reduce(), seeded_f(0, err=0), atol=NOT_CLOSE_TOL) seeded_f = apply_seed_to_func(f_non_lin, SEED) fac = PolyFactory(X_VALS, order=2) fac.iterate(seeded_f) assert np.isclose(fac.reduce(), seeded_f(0, err=0), atol=CLOSE_TOL)