lot

tests/test_generation.py

@@ -0,0 +1,20 @@
+from untitled import generation
+
+
+def test_cat_description_gen():
+    traits_a = {
+        "size": "tiny",
+        "color": "tuxedo",
+        "eyes": "blue",
+        "personality": "judges you silently",
+    }
+    traits_an = {
+        "size": "average sized",
+        "color": "tuxedo",
+        "eyes": "blue",
+        "personality": "judges you silently",
+    }
+    a = "A tiny tuxedo kitten with blue eyes who judges you silently"
+    an = "An average sized tuxedo kitten with blue eyes who judges you silently"
+    assert generation.generate_trait_sentence(traits_a) == a
+    assert generation.generate_trait_sentence(traits_an) == an