test_example.py 文件源码

def test_params_regression():
    """
    Test for regressions in model parameter values from provided data
    """

    model = mli.Model()
    ortho_x, ortho_y, ortho_n = mli.transform_data(op.join(data_path,
                                                          'ortho.csv'))

    para_x, para_y, para_n = mli.transform_data(op.join(data_path,
                                                       'para.csv'))

    ortho_fit = model.fit(ortho_x, ortho_y)
    para_fit = model.fit(para_x, para_y)

    npt.assert_almost_equal(ortho_fit.params[0], 0.46438638)
    npt.assert_almost_equal(ortho_fit.params[1], 0.13845926)
    npt.assert_almost_equal(para_fit.params[0], 0.57456788)
    npt.assert_almost_equal(para_fit.params[1], 0.13684096)