stuff

tests/test_generation.py

@@ -1,4 +1,4 @@
-from untitled import generation, rules
+from untitled import PACKAGE_ROOT, generation, namegen, rules
 
 
 def test_cat_description_gen():
@@ -21,7 +21,12 @@ def test_cat_description_gen():
 
 
 def test_generated_names_pass_filters():
+    names = []
+    with open(PACKAGE_ROOT / "assets" / "cat_names.txt") as f:
+        for i in f.readlines():
+            names.append(i.strip())
+    name_generator = namegen.NameGenerator(names, rules.validate_cat_name)
     for _ in range(10000):
-        name = generation.generate_name()
+        name = name_generator.generate_name()
         assert rules.validate_cat_name(name, auto_gen=True) is None
         assert any(c in "aeiouy" for c in name.lower())