def test_save_load_model(self):
base_fitter = lt.BaseFitter()
base_fitter.save_model(PATH_SAVE_MODEL)
assert os.path.exists(PATH_SAVE_MODEL) == True
base_fitter_saved = lt.BaseFitter()
base_fitter_saved.load_model(PATH_SAVE_MODEL)
assert repr(base_fitter) == repr(base_fitter_saved)
os.remove(PATH_SAVE_MODEL)
def test_unload_params(self):
base_fitter = lt.BaseFitter()
with pytest.raises(ValueError):
base_fitter._unload_params()
base_fitter.params_ = pd.Series(dict(x=12.3, y=42))
npt.assert_array_almost_equal([12.3, 42],
base_fitter._unload_params("x", "y"))
def test_repr(self):
base_fitter = lt.BaseFitter()
assert repr(base_fitter) == "<lifetimes.BaseFitter>"
base_fitter.params_ = pd.Series(dict(x=12.3, y=42))
base_fitter.data = np.array([1, 2, 3])
assert repr(
base_fitter
) == "<lifetimes.BaseFitter: fitted with 3 subjects, x: 12.30, y: 42.00>"
base_fitter.data = None
assert repr(
base_fitter) == "<lifetimes.BaseFitter: x: 12.30, y: 42.00>"