tests: fix req. explicit field in ImageGenerateJobPopResponse test

Haidra-Org · Feb 1, 2024 · 5b00204 · 5b00204
1 parent 4f7148a
commit 5b00204
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/ai_horde_api/test_ai_horde_api_models.py b/tests/ai_horde_api/test_ai_horde_api_models.py
@@ -328,6 +328,7 @@ def test_ImageGenerateJobPopResponse() -> None:
                 post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus],
                 prompt="A cat in a hat",
             ),
+            model="Deliberate",
             skipped=ImageGenerateJobPopSkippedStatus(),
         )
 
@@ -338,6 +339,7 @@ def test_ImageGenerateJobPopResponse() -> None:
             post_processing=[KNOWN_UPSCALERS.RealESRGAN_x2plus],
             prompt="A cat in a hat",
         ),
+        model="Deliberate",
         skipped=ImageGenerateJobPopSkippedStatus(),
     )
 
@@ -351,6 +353,7 @@ def test_ImageGenerateJobPopResponse() -> None:
         payload=ImageGenerateJobPopPayload(
             prompt="A cat in a hat",
         ),
+        model="Deliberate",
         skipped=ImageGenerateJobPopSkippedStatus(),
     )
 
@@ -364,6 +367,7 @@ def test_ImageGenerateJobPopResponse() -> None:
             post_processing=[KNOWN_FACEFIXERS.CodeFormers],
             prompt="A cat in a hat",
         ),
+        model="Deliberate",
         skipped=ImageGenerateJobPopSkippedStatus(),
     )
 
@@ -377,6 +381,7 @@ def test_ImageGenerateJobPopResponse() -> None:
             post_processing=[KNOWN_FACEFIXERS.CodeFormers, KNOWN_UPSCALERS.RealESRGAN_x2plus],
             prompt="A cat in a hat",
         ),
+        model="Deliberate",
         skipped=ImageGenerateJobPopSkippedStatus(),
     )
 
@@ -392,6 +397,7 @@ def test_ImageGenerateJobPopResponse() -> None:
             sampler_name="unknown sampler",
             prompt="A cat in a hat",
         ),
+        model="Deliberate",
         skipped=ImageGenerateJobPopSkippedStatus(),
     )
     test_response = ImageGenerateJobPopResponse(
@@ -401,6 +407,7 @@ def test_ImageGenerateJobPopResponse() -> None:
             post_processing=["4x_AnimeSharp"],
             prompt="A cat in a hat",
         ),
+        model="Deliberate",
         skipped=ImageGenerateJobPopSkippedStatus(),
     )
 
@@ -411,6 +418,7 @@ def test_ImageGenerateJobPopResponse() -> None:
             post_processing=[KNOWN_UPSCALERS.four_4x_AnimeSharp],
             prompt="A cat in a hat",
         ),
+        model="Deliberate",
         skipped=ImageGenerateJobPopSkippedStatus(),
     )