def test_with_boot_with_jac(): f, ferr, ferrt = fitfunc_to_func(ff, [1, 2], [[1.0, 1.1, 1.2], [1.9, 1.8, 2.1]], jacobian=ffj) assert_func_equal(f, lambda x: x + 2) assert_func_equal(ferr, lambda x: numpy.array([x + 1.9, 1.1*x + 1.8, 1.2*x + 2.1]).T) assert ferrt == 'boot'
def test_with_boot_with_ff(): f, ferr, ferrt = fitfunc_to_func(linear, [1, 2], [[1.0, 1.1, 1.2], [1.9, 1.8, 2.1]]) assert_func_equal(f, lambda x: x + 2) assert_func_equal(ferr, lambda x: numpy.array([x + 1.9, 1.1*x + 1.8, 1.2*x + 2.1]).T) assert ferrt == 'boot'
def test_with_std_with_ff(): f, ferr, ferrt = fitfunc_to_func(linear, [1, 2], [0.2, 0.1]) assert_func_equal(f, lambda x: x + 2) assert_func_equal(ferr, lambda x: numpy.sqrt(0.04*(x**2) + 0.01)) assert ferrt == 'std'
def test_with_std_with_jac(): f, ferr, ferrt = fitfunc_to_func(ff, [1, 2], [0.2, 0.1], jacobian=ffj) assert_func_equal(f, lambda x: x + 2) assert_func_equal(ferr, lambda x: numpy.sqrt(0.04*(x**2) + 0.01)) assert ferrt == 'std'
def test_no_extra_with_ff(): f, ferr, ferrt = fitfunc_to_func(linear, [1, 2]) assert_func_equal(f, lambda x: x + 2) assert ferr is None assert ferrt is None
def test_no_extra_with_jac(): f, ferr, ferrt = fitfunc_to_func(ff, [1, 2], jacobian=ffj) assert_func_equal(f, lambda x: x + 2) assert ferr is None assert ferrt is None
def test_basic(): f, ferr, ferrt = fitfunc_to_func(ff, [1, 2]) assert_func_equal(f, lambda x: x + 2) assert ferr is None assert ferrt is None