def test_conf_int_raise_error_missing(self): fm = FittedModel([0, 1, 2], [3, 4, 5]) with pytest.raises(RuntimeError): fm.confidence_interval() with pytest.raises(RuntimeError): fm.default_confidence_interval()
def test_normal_run(self): fm = FittedModel( np.array([0, 1]), np.array([2, 3]), conf_int=np.array([[4, 5], [6, 7]]), x_all=[4, 5], y_all=(6, 7), w_all=[8, 9], ) fm = fm.prepare() assert fm is fm fm = fm.fit() assert fm is fm np.testing.assert_array_equal(fm.predict(), [2, 3]) np.testing.assert_array_equal(fm.confidence_interval(), [[4, 5], [6, 7]]) np.testing.assert_array_equal( fm.default_confidence_interval(), [[4, 5], [6, 7]] )