예제 #1
0
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)
예제 #2
0
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