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]]
)
