From fcaeb8cb42fbced130fdfe13fdac56da564bfbd1 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Fri, 30 Jun 2023 22:32:20 -0500
Subject: [PATCH] fix(api): add non-square txt2img test

---
 .../test-refs/txt2img-sd-v1-5-tall-muffin-0.png     |  3 +++
 api/scripts/test-release.py                         | 13 +++++++++++++
 2 files changed, 16 insertions(+)
 create mode 100644 api/scripts/test-refs/txt2img-sd-v1-5-tall-muffin-0.png

diff --git a/api/scripts/test-refs/txt2img-sd-v1-5-tall-muffin-0.png b/api/scripts/test-refs/txt2img-sd-v1-5-tall-muffin-0.png
new file mode 100644
index 00000000..b7d4f956
--- /dev/null
+++ b/api/scripts/test-refs/txt2img-sd-v1-5-tall-muffin-0.png
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:084c489e1e4d374c59c8a4a0ed476805815d3602732fe37f66032473cd4595fb
+size 794421
diff --git a/api/scripts/test-release.py b/api/scripts/test-release.py
index 2db2fdbd..aaf072e0 100644
--- a/api/scripts/test-release.py
+++ b/api/scripts/test-release.py
@@ -307,6 +307,19 @@ TEST_DATA = [
         "img2img?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&sourceFilter=none&pipeline=panorama&tiledVAE=true",
         source="txt2img-panorama-1024x768-muffin-0",
     ),
+    TestCase(
+        "txt2img-sd-v1-5-tall-muffin",
+        "txt2img?prompt=a+giant+muffin&seed=0&scheduler=ddim&width=512&height=768",
+    ),
+    TestCase(
+        "upscale-resrgan-x4-tall-muffin",
+        (
+            "upscale?prompt=a+giant+pumpkin&seed=0&scheduler=ddim&upscaling=upscaling-real-esrgan-x4-plus&scale=4&outscale=4"
+            "&correction=correction-gfpgan&faces=false&faceOutscale=1&faceStrength=1.0"
+        ),
+        source="txt2img-sd-v1-5-tall-muffin-0",
+    ),
+    # TODO: non-square controlnet
 ]