From 14efe02544ce418c475bc12f430cd5bae7ca2231 Mon Sep 17 00:00:00 2001
From: Toasterkitten <owen@ericfeldman.com>
Date: Wed, 24 Jun 2026 15:46:46 -0400
Subject: [PATCH] adoption flow

---
 tests/test_persistence.py | 23 ++++++++++++++++++++---
 untitled/screens.py       |  2 +-
 2 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/tests/test_persistence.py b/tests/test_persistence.py
index 7d30cb3..34de4c5 100644
--- a/tests/test_persistence.py
+++ b/tests/test_persistence.py
@@ -2,8 +2,25 @@ from untitled import model, persistence
 
 
 def test_save_load_roundtrip(tmp_path):
-    original = model.Save(version=1, cat=model.Cat("Mittens"))
+    original = model.Save(
+        version=1,
+        cat=model.Cat(
+            "Fry",
+            {
+                "size": "tiny",
+                "color": "tuxedo",
+                "eyes": "blue",
+                "personality": "judges you silently",
+            },
+        ),
+    )
     persistence.save(original, tmp_path)
-    loaded = persistence.load("Mittens", tmp_path)
-    assert loaded.cat.name == "Mittens"
+    loaded = persistence.load("Fry", tmp_path)
+    assert loaded.cat.name == "Fry"
+    assert loaded.cat.traits == {
+        "size": "tiny",
+        "color": "tuxedo",
+        "eyes": "blue",
+        "personality": "judges you silently",
+    }
     assert loaded.version == 1
diff --git a/untitled/screens.py b/untitled/screens.py
index 1a2803a..766d2a0 100644
--- a/untitled/screens.py
+++ b/untitled/screens.py
@@ -1,4 +1,4 @@
-from untitled import generation, ui, rules, model
+from untitled import generation, model, rules, ui
 
 
 def adoption():