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>"