From 489b5d5c235b7d468aea37256615e1c51c40a484 Mon Sep 17 00:00:00 2001 From: Toasterkitten Date: Mon, 22 Jun 2026 18:04:41 -0400 Subject: [PATCH] test --- tests/test_persistence.py | 4 ++-- untitled/persistence.py | 12 +++++++----- 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/tests/test_persistence.py b/tests/test_persistence.py index d817cde..7d30cb3 100644 --- a/tests/test_persistence.py +++ b/tests/test_persistence.py @@ -3,7 +3,7 @@ from untitled import model, persistence def test_save_load_roundtrip(tmp_path): original = model.Save(version=1, cat=model.Cat("Mittens")) - persistence.save(original) - loaded = persistence.load("Mittens") + persistence.save(original, tmp_path) + loaded = persistence.load("Mittens", tmp_path) assert loaded.cat.name == "Mittens" assert loaded.version == 1 diff --git a/untitled/persistence.py b/untitled/persistence.py index e9c1516..6bc1762 100644 --- a/untitled/persistence.py +++ b/untitled/persistence.py @@ -6,21 +6,23 @@ from untitled import model SAVE_FOLDER = os.path.join("untitled", "saves") -def save(save: model.Save): +def save(save: model.Save, folder=None): + folder = folder or SAVE_FOLDER file_name = save.cat.name + ".kitten" - save_file = os.path.join(SAVE_FOLDER, file_name) + save_file = os.path.join(folder, file_name) data = save.to_dict() - os.makedirs(SAVE_FOLDER, exist_ok=True) + os.makedirs(folder, exist_ok=True) with open(save_file, "w") as f: json.dump(data, f) -def load(name): +def load(name, folder=None): + folder = folder or SAVE_FOLDER file_name = name + ".kitten" - save_file = os.path.join(SAVE_FOLDER, file_name) + save_file = os.path.join(folder, file_name) with open(save_file) as f: data = json.load(f)