def test_save_history_no_dir(model_name: str, tmpdir: py.path.local, new_history: History): shutil.rmtree(tmpdir.join("models")) save_history(new_history, model_name) assert tmpdir.join("models").check(dir=1)
def test_save_history(model_name: str, new_history: History): save_history(new_history, model_name) history = load_history(model_name) assert history == new_history