From 2f3d8172c3fe80ea64ebe5c797835ca15c2e595f Mon Sep 17 00:00:00 2001
From: Vladimir Repin <32306715+mezotaken@users.noreply.github.com>
Date: Sat, 29 Oct 2022 21:43:32 +0300
Subject: [PATCH] img2img test template and setUp added

---
 test/extras_test.py               |   0
 test/img2img_test.py              |  59 +++++++++++++++++
 test/test_files/img2img_basic.png | Bin 0 -> 9932 bytes
 test/test_files/mask_basic.png    | Bin 0 -> 362 bytes
 test/txt2img_test.py              | 102 ++++++++++++++----------------
 5 files changed, 106 insertions(+), 55 deletions(-)
 create mode 100644 test/extras_test.py
 create mode 100644 test/img2img_test.py
 create mode 100644 test/test_files/img2img_basic.png
 create mode 100644 test/test_files/mask_basic.png

diff --git a/test/extras_test.py b/test/extras_test.py
new file mode 100644
index 00000000..e69de29b
diff --git a/test/img2img_test.py b/test/img2img_test.py
new file mode 100644
index 00000000..d8ed309d
--- /dev/null
+++ b/test/img2img_test.py
@@ -0,0 +1,59 @@
+import unittest
+import requests
+from gradio.processing_utils import encode_pil_to_base64
+from PIL import Image
+
+class Img2ImgWorking(unittest.TestCase):
+    def setUp(self):
+        self.url_img2img = "http://localhost:7860/sdapi/v1/img2img"
+        self.simple_img2img = {
+            "init_images": [
+                encode_pil_to_base64(Image.open(r"test/test_files/img2img_basic.png"))
+                ],
+            "resize_mode": 0,
+            "denoising_strength": 0.75,
+            "mask": None,
+            "mask_blur": 4,
+            "inpainting_fill": 0,
+            "inpaint_full_res": False,
+            "inpaint_full_res_padding": 0,
+            "inpainting_mask_invert": 0,
+            "prompt": "example prompt",
+            "styles": [
+                ""
+                ],
+            "seed": -1,
+            "subseed": -1,
+            "subseed_strength": 0,
+            "seed_resize_from_h": -1,
+            "seed_resize_from_w": -1,
+            "batch_size": 1,
+            "n_iter": 1,
+            "steps": 3,
+            "cfg_scale": 7,
+            "width": 64,
+            "height": 64,
+            "restore_faces": False,
+            "tiling": False,
+            "negative_prompt": "",
+            "eta": 0,
+            "s_churn": 0,
+            "s_tmax": 0,
+            "s_tmin": 0,
+            "s_noise": 1,
+            "override_settings": {},
+            "sampler_index": "Euler a",
+            "include_init_images": False
+            }
+    def test_img2img_simple_performed(self):
+        self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200)
+
+    def test_inpainting_masked_performed(self):
+        self.simple_img2img["mask"] = encode_pil_to_base64(Image.open(r"test/test_files/mask_basic.png"))
+        self.assertEqual(requests.post(self.url_img2img, json=self.simple_img2img).status_code, 200)
+
+class TestImg2ImgCorrectness(unittest.TestCase):
+    pass
+
+if __name__ == "__main__":
+    unittest.main()
\ No newline at end of file
diff --git a/test/test_files/img2img_basic.png b/test/test_files/img2img_basic.png
new file mode 100644
index 0000000000000000000000000000000000000000..49a420482d0a70b9f5986d776a66cb3ea39d1a97
GIT binary patch
literal 9932
zcmV;-CNtTIP)<h;3K|Lk000e1NJLTq002M$002M;1^@s6s%dfF00001b5ch_0Itp)
z=>Px#1ZP1_K>z@;j|==^1poj532;bRa{vGi!~g&e!~vBn4jTXf02XvbSaefwW^{L9
za%BK#X=XBTZf77eE;KGOqh3c2001UGNkl<ZcwSAL3A9~jb>H_r=gz~MC(X~YJV>_j
z0N5CuKm%P6%#>wTlQvx~4yA2Eikl8$B~7W;YFA5}(u9?)5axCfLJ6ca1``j&#<mQ$
zv5hR-k|j&FWDW1>=}q^&cjq(o_uuD=^gi8trtf@v-uu75bCmnaXI@QHOD&f({hEF)
z%~`-U;pp-wU_ip9Y~L(qtI>g<@w@z!V^8zsm^G(?nvQRdf!Tb2JAW#od~$xX-{sqc
zO^z`>mk-JI^-rH{WOkrAoR*)@zM_-M1p3)WyUvbwhJ*dyFaP9}-yU+(@8N<I)&L)w
z-H_-6GY1SzI3WRsTH`peP=m667WpNONi0T!Jo{-jG4o{mV#Ys6f0?md8n7QbV;uS?
z<U)-0{C-W~7k<W>^P<uE)mFmGP$84Xnv>(~PVp^+4d)qN$o<bJPNgwNVnRat#1d?H
zEE8N7HWNbI;msPxj|iqYSdnm?3m6a$=e$f$o{}+vz-(h~3Wn|z>c{D^;Tr@pfJW<f
zc=BsH(ibr}aoOtf#1xmwbC32lPS-*Vr4gAD*yFtT8Tjo(ruY}QHMt>$x`fdseO+D7
zgtV%OvYRx{KM>RqXM|H}#+ToGS@Syo|GTGaj(-UQ{EuVJp}BopU1qR`$#ZmGb43W~
z?1%`nT@D|O(cB@5l<>M)bBdzo)4e_!w`tzK_J>N>Oi~s~fry(&H?v{ix4PM<(k8Q-
zgwbhzT{yO4Qh&-3I@Wv&2f2MBR^5HM#tmY^=Yxhq?q^nJn<*ljTMCzIvTb%Q3xRA*
z#CWI6SI3!h7|Ud`aw-p!&}czk>~{|7x!^kFSYJGV9lzjYhJ$cnF6!EV38YO;Ghh}@
zJ7M>np*JK5r`-fgr9r`RrJ@xIIh!bSZM2kHrIg35IiCw*QgdkhP&j@9X;V9*;&bz1
zoKA^J4ZO^}8%D<*$XKob|7L=^BG~b8;Oq0Si8WpVI~_i3+&jPW1XX|wn%HmPiX1|D
zL9}&a$i)EFDsG)0jpt8<rST+}w_+i0)u3<Hjn^$YbIe*tk67FrSYvX`N;5OIF>{rr
z(>p8<$E_y{5Rc$q^D3O28t24=Xn0<LPz-Gs+7*)V=JZ9Jb5elDYnsc#x{SJyzu_>x
z*?#uPu4!EED?fS_Pq}<Wv*=fUF*iUiqwYj~GJSOhiki4wELx>|(ORdDSn<@0wsGNt
zH8=2~D6vW&hQp$DAQ&Qc%>LMiK4Q6@`z?+mcTrK22e`Wr=XPNvivME_g!JcOz_siy
zl*F+-RUs!&)p^~Ug=$vW2HF$zd=$p13y>Xi$#P%911?h@$~x2(gMCCck%ss=9~8J(
z6iV3Xnfq+(&<|~+o3ma&v2s4Jsqvx}xhX7^tdAGSnla(A+}yI!fBHKXHYNy{$j^z2
z_>miCRwS<RU(t;D%Hm=}bBMNlz&UW9GsVekLWLj4J{KoaCX*bx!^^NwlWlchFdVN8
z#ZGoHVW60jpk69C&T|wEeJ9Hk2<Cek!>du%E?sI7>w(Qrmu+sQ>hT`&Ro0b@_XY#X
zVr=7+4a-%kkR-t|Rly@6v!*z3hGD)CW;_uYZaudo`yR*8_Nt|u#Ev-8eaxj?<RRgj
z<bDaLYkW0_pVM`|(rk_-f>&f$d=g!h$~koz6GUt=>QHnD!=<js&G2$-%uL$c^oW(J
zWwU(I`rXK)LErlQfhE0|5Xf6tsxUsWzGi@cXo(TctrDiWL-^4AQoNF>By`v3Sl9EM
zk|p&gL9?-3U=Mv|!Y0Ac+QUCTFPg(#=4V|zzZNGblk<Kc4gT!I328@4P~0RgG=rE-
z6EYwhCpY3+uHR+0$c2`0MG#H}D@Z4OD2L>NO0i%uipYlrzgylGz*&M~bdP>yB^S}R
zoOJp_+Rjtvg&O1W>{~8zUQ=WfWt+UNaYJ^mPU>ErlTv(`Uq^91#0O>DY)69W12BaE
zBy~=woR}ErW>lXWv@OmTm?$9bv37wWm@AhLEnltLMpCqAR|@uQGiNU^^<g5xTdGzp
zl^7dxAJHv_G&f%fCreR9t$^!^w&M!8mN#hMVYj%DQZkkd1nmlPX|!c*rXpdIY;Mgf
zqbf|K1Q|bk->=<s(2dUtX+$SAk)Lw)a^5#g#g2&jSncP@5a(a9<kb@v6pAP<q5i`(
zSuB<fNOaS?Chf+L-fhblRxDXxwo5NRZOgBpu>R&n3%I>dsd*(Smdhvv1xipRmsQ)X
zH7n*zZfzyROdK&L0i_T>sec$F6G%s{APwZYDu>_4u_8+|DBM)iJ?y@4=oAT4S7m8#
zmwY-N3aUxPK=#SJL9Q#URLY@so_^SZC+@SLh(+^dgn}HB^*aO0lNd{r)An~h^?SD2
z?%FFS&e*dLK5Q4Behwvt2pm~yeAIfm$m-Q`YwXx&rHLIjGB;&WrEGuk&wj&x<?~;)
z`t(jPMGqDu4`w;1TTlFmEB#0qHf}(^`jr<5M!S6UDvI+=R6>g#CHLGHT>(l+nlSrZ
zb|d|GgF~FASVgDw^$GbDD^&L%-EYwo4_a+xjL8|J$#8$wvOFlTv%YD6{?xMli%);t
z-hStMY;B`uM_xW*Pu~B4Z5=*|YnN<nV$?P^*O@nOQPj1RT#^>^)-G3UqTaB^u5DJ`
zeYK76p10!2j5CqQ%ObwYZW3F<a9*5U?nd>IyDPWgoIrjVFB>D}$l|%r(+x_6a)+VO
zT>&{II0j|z<G((kHWE_ht6a@n@3FtK;F-s9<*F_a7DW`<2NPiZVru15V&8n=HGA~M
zv-XQ0yv6?EzxfTj`Sx3FeRIox^5O}5=C2>J&asmq$%2(Cb<kmK)oRf;w+2?O=Pil)
z)?M$yU*w6ipv+3xjPJYB>f3f&bvsI_k3+ZQd`ck+(xwSOPW;MUGFjPGKFDN}V$@F%
z0w+@muW`~qmhduwA!AA@F1rM0Uy2P8RHsW-Lf{AAvf#NNS-#e=gg`7vIn=fo3Y9W-
zHXYc}*VgR^&o0_cAGpCfi)%J_<zD;HKl*!i)q!gPN1OKCi!a;Z`yRCRYYTWp2zm=_
zY@}$tPS3#<1;jd=C;=r6`W*zs^SV6}X$sH+v=>sVPR?0rdd9}~?6#1Q8JU=}peC;*
zP$UFZ%JGaa2)s)G9TEa3<p?^(g$y`Ic>QKDLCVlao{;4OUDa=~9$M?ZKePPN!&a&l
zu|gTwB*~}}K#-=~fz+i|&mMSs!FKH5X-lsy+RD<pwfa4K#|M7S{_!V&-R5BW+@*DU
z>Bvz#{@5YwzIFzhYN)uI$`W8vRii;avb@j}=hE?t1sB({#S$2lRZ~Ku+ieqpR2M2#
zfKkTgX01B2-NJ1<Y;=0o!m&vQY7)kKn=g?AG6$DyNH9caLS=jhL!1!M@G&lVR;Dzd
zP|I8AzOPwu?1&XcYF@fxz_~m}2}QXh7XgLo2EX<2>sSmA5n-;idyzHQ+tvXkzW49H
z$L_l4ZkwK+wR20W_WZLywbu?kVOysc$uni<AVkX~S(0j}9aAX~u#9q;BTWbqU?OJ-
zp^|%s6qizgWUh9%W3g625O#D7Xim;Cs#kRz33m#`eBkbT4k|t*q7)z%WZ6|GGu6oT
zl__08K9(PL3NX}v`LI#rtXu)@2^ylke2#ig4pA)TuvQu7mx*avuu!IXBEi)v1)8PM
zWB>3IFWLQHzZaL^us7d!qaC<zzcuz<YsHxfTRVTox*Gs3xDqZH^kW+gdRB$0T!{N)
zb(PRAicC;;xgj(P^hV$UbX7*n)dFJ(sXnsEn|ggi`__8>w1pr1C-)qbXztps2^TZl
zVr7MdTa%5(=p%d|2&4?U*8cGWR>A@WYWV;L<TWWV5MzNnW7M%JraK)hMYB{z_yPb(
zt*}C?JW{aHM%CNfXC66h-~ak|u(a7*-gb*!b<LGF|HfOaNg7{%{jBwyJ*yFRO2G2c
zBA(Uirzoy#BO?`*fc|N|<N<l|5ag|og8GD8f#e)e$+{d@iXa6_wd@tX?<1c$D4+FS
z8~Kz3FMm`Pb6!LE@{jqrkzXkQhTG>~wAqbii<`^VBKFk*ma0QArH-x&h`|LzTvM3g
z>_Qu_$&-D3gqNp0!h}FICDxHqatKO#`oX8{2jBQBo7g^KH@)d5+jsSqw(Z92ty45x
zJoUQedR<S{$ZoV&RvD`xBo&9KNbm%s?N-mBrMH1{5+M^)U3}pMx^Ws>3Gek1`M!^S
z;-D-l<7d|u>sdIc>F^zwUIP^0d?CL0+VghToo~0ZPm;9OTe!4{W|l7ijj@j-!jOd2
z$K}w!?Vc{$Qj?y6n8XRKjxrtLVTzLIqKI;-Xti1iJdoN0-+0gteg8q5-#>3RzVRlT
zzjEGYZoJMGNvfOY&KXtKViM1whY|?EKFZ6N-~q{`P%WnPJ016k240qz@(17(t%S;m
z;EvoEA3F)gxtsze>Tn++eN~8E!A#)-d4m+tJoT)7@ZE2*!_PcpCqMU9+qtblyGydl
zqve?Ri%<d~5tEPtYPPJU`ddI5aT?p{<$;}CY&nxfxhWpegXsW843xZ7x^UIjAhPym
z&(4z=Z+*v&_FMns-&tc~8n*{ko?Eu(zV&U3*H$f}TncxH-!cl3A|e!CAlyno^d1J#
z5`AN=YHbi|OdiQAA(Ml0UlI_)ofByU!2{r81VeI9@73iE?j6K4L3Qcm5qsay@3)H!
zOZNL8`?vPtTdu@Y&^(z{$xU5lOlFQStuR53YjKKi0%%GoKyyf0J=i^VwrT6DB)Fk+
z<Y5w;!1aaN2u~4M>C$4KyRc?&d)FK7Uw!4%V1hQZinh|+uv5ogu^dVc3B}gRCEHxv
zvgYc#b+ujF@RBHxkWwp^ycLU8>kvW%lwBh<b6<M=WLE1SD7Y;9dxUb2kQY~+F(s~x
zXO7!D->`%D?%RL9=eI523+=sEPiq7kWir)Xte%2gdRVYTD5xLk_NeMoBFQMm?E*5t
z=1>E**@^7f>2=!z&=jGe9F%Zx2L@x5AxxsR#pYJmcI@3@2k-r&4Z2+qmHY31$X<T-
zuqVSP%G+L&rYcRjTDOsU$?{Zz+*Z?;F;=J5vi@q*E-j;+jgEJSy#XHZMLa;6S^=P<
zCu>Us+A0Z3F?mYa@|l<I)~ly%cB*cl{`l`u$2*p81$O7vqwZ=7qX^eb)ew<^Q<9Ow
zhBA(eBbaKpp8BK(<|PlPOi`l3;S$ZciyeD)x#J5edZ^(rRbfcFAK(cc-24i1!TUe?
zF?;^dQ+Dcwqqcf!8PCD%C~<|fWWW6fpR(`%<-N9aj<QT2p%tkdb-!9K1F!3##Yr2b
z0z_1RQom&@ix;yH5Fxlg0a#yF=g`3}F{&iCx^%&=pX}S!`}f%&-t#GLT%dkc?Beqm
z?e07GI{x;)-Ww*}L#<Xq;#BNUEmG<+xL*5sxRjBQE96<KDx@uxA@Gve(F<*qu{q>J
zy*oUCDoZ4?nDAL$-n1s2Azn;~&};#QO3-`n-S4*jcid**{p0^dWx@MKYgVs~T1a_R
zK2uH#2!R~XI>2C-BiWABE%#-RpdhfUFQMiBT1^pi0TQgPUb6YzioNaUZ?@0;%l}{t
zubu&5RP5~0C7YmW`Q^9ISqt<hs|9NCvFIQ|B)|$ZeJ&*@c~eX+*0U)Angk|Q8`7v_
z;1uYphGBh_u5dVf4i9UTthQ%|jcuE=(ebLCe)=U_c;*B;RP7&-_+ndG>)1Pf?VZdK
z*kgbB81Bw|^bsL0T%ucTjFc&PfUXe$5lRf%mojq1f2j(1`&$pt2O*Fh?*z|>)?Qz?
z+U6_v{tv#({@|1U$xc7_k`>2C?EDLh)<BE(TG{TreVUvl1z;foAq4PGzwcdLkC+s6
z2-P6KR7{tq)gra_Im!`w04T{aT0#L?lKpyHZ9Bj73Y)%SJFZU>Y_D6h4?l-ruxMe`
z5-BhVXrWs+d(AGZUomAr{<9xixmflEl@d89!kYxPA|XeWML8%Qs8U-cWb(Cw)c_7M
zg$wK+?}n#kgunm+w$4!Q?s~6%?f3u0UVY|8tIdqt(o0L$C>KCpxDchJ&*TI8_lW!C
zwo+o|^)x0}djO*mAjKEMIHxMm<9Gwi@On_OGu!6v?2g?qSH)|%H*Q;%F}J<_4m){n
z#g3miZcA<2<XqiudH*}C^rm?`dU}-#gD1iB%+-5s|J&bXH+<-=mLDzIzMJ=2mBc$%
zFWMx>6XRu@o-EtcWX(o7SHK8btm!PZZRzzU3gF|bt=jYS53oQZtFIlgkACDX`{tLv
z1~PlxMz>Ab!b^)*BZ<_AC0XuP5+G^@c;Rv+Mfk?l2#r!2((({+DE14bGUYXAPb|iE
zZu=fPS!r1Qz#Hw%#T7f3PE#q==HeZRTHb+Oam{WU-F}4?Yf~21sTx%puX7h|CN10i
zR5fE2f-@6i0I>}__cuSauoc<Ls~4zP1*fmY*=a(cLHLvjkqQ87da7nqAjiq^5)hoc
zpt7ULPgTH);{oVx<>Yhr%Xi*phwgjG9{<+&ZG6v+oj$f;WyTbNv_)L_)&nqx3&q1l
z%>V)c^}k9SSD2?y3DXKuEg-I$a{yk?gYIs+?K-=F(9eJSQQI+6wUOOd!33p?IfJA}
z;_JAp7U>6eZJV@Q9VL^M1#BW9_bX4mNO`56!*_vkNkdf!qd;?RvTCD*U76NAAr}=<
zP@i#SlC*%BdVmt}yhv5ZGrrQO5%Rc^9)S%Jysf?Z0&#r}@l>(n=a+2nZ8vz2-vC0^
zv4j>ZZn<*I#wqLx6PJi7Z-dF@0c9;kA&Ni6K|s>cBP(UyfE+L&-Cl*u&g|N5b64%M
z=^axxIx}l2$`k!O-jix^6~(2LuU$K4tU5h|w~4o`<*lo)AoqA##b!Z4jTFEe8l`%L
z+F$gBt6Z-lEXn{c<l(ViE&Ed|Ey`E4WgHEuV7+QH85tsDVb|V$$F<g&obpA6_uchN
z#Mm2b+m(B$t0EZKo30sW(wzMb<?2Fx3hlu(R?jKR5IWH?$tl?l+8sZqQd&^f!_AAe
zhD}UOTS9rtH|jP%Icg<xN+6HHb@i@E3V>35S27zEYgXQNg<DzWu};ajYWt|&e(i`&
zGzvC`MXO~RF?kSb5~8KRDy2DH;fRR20=&jou`!5R`wEIw$H=Z+AD2_skMq{-=IrSo
zJ#C-+@Bh{Q;!i(oCx85eEx&w*=m>4!j)qOrzJ@4k_nY2eD<C{-9>S5}Kx+d4kBYxg
zUJ=3thzRwmzKbX$m2kl14GSh`No09|53MQ|>tc0zS^@%eScX^0@<o~jtpb&1cUc4e
zw?Ti~7@<W19-kVw*@>FXfbmrt)ua{4O$C&$1SusbR7zM|A3f_39#KiwM+$>2ZbS}U
z?8TuSUn|+h*!8yWJs-9CU-+;cdH5;2_`+$sX3v<-&sMzWkICS7+;J;R2su!41Z`!-
z9OzG20F4hoggwB9*wE`UF)mwBAC9sWB86%#V+NNW0i~4)qFCq=CCGBpK$k903m6il
zgpzhoS8e+kF^?y<X(wkZIg(k!dTVRMHH|SL7a>)`{e1wEdVqvj&ru59)F?y1Ol3a<
zdYlvFb)5?i5ej;kA)Kt|eATfz``{;k(+=Eztxe9<Z2$q)axF=ZZlANt=me8c`xU>6
z;(Uz`5W=l4S_U2B2{8g{u|qK)kw-!`E+}kv5*Un7>c~>iS={ZT95NrKlK`qpN>F+Y
zM#kYEf^<Bzx|A{TS%uL6Fw-C(y>tJBR=Vu!?KOMz-nw14chq)}>E}iPLMS0oUm-pm
z^Ki^YUsr_ULTF9@^pRKORNtxNRV-Sb-)Rv&c!(ykohLmds>$hbCM9+gG*6=w&<m7s
z*R((bAZ3XHKq#e5N@-G_%3`c%i9zO;yC&=Q=4-cWc}LNN+p9B#2dty@LiHHh7i%3n
zw3|~{K?4uddISFLWCn!NtyhfL^cbE6#GRgQ*fsOxcI}=~yY;#Wd+YU6cH6aMcEg^A
z?b%KqB6ModaSzZ(7R+|lh-ADgcFWtgTloADpr*=>o)pu?q4@+i99ce(99S4-gj%nc
zl_N@8jxk{U+yFoAw#`~wS-hoPM&#`F+itU8{m6$cUmvr%opV;&T(N$;1=vxxPr&(l
zs!YycE-|HSFMdMhDtM4T_1zo9FCXXx#JR$SfVBSlmUUYlYk><==7}+g7EA?+M<yG#
zmt1!96=U|co44a-%&eFa7LWq;%^rwk2dNs5jh45#5$6-P96rK;E=<P&mmtQX)cHcK
z0s~}!u4^SWNr~}V-P*LB5uVU^^IPqe)s{W+zz^(ufBaSZ=I<S}Z~x}Mw!?q)Ijd9~
zzL+7Jl|~f>;57&&v#>$Q2uFXBlVS;1qQ}=QwD1{|H<V)r3L(O$4){@7EK(f?C_(MD
zx=fYm5|a@@w!!gBD|$SxG$=x{xc~9d>W{?4^?Ds4G|G$?`M?Y6UX`QtNIY3MD^Z=W
z9)Kv0^$1)|k-EElK^94^j1^RO7oL0C&V2T>W{*5%+b$lr{e`Ywx1BhhuE8W=nFN~x
z!DA|cYl4qBQlpg}WXMvvAtfABUoI~LvM>;Qxn~MPXdM#w@P$!4rm6-F8MnHr9Xj8!
zr}#c{KDNau!xFA#xv_*-{5x%rJ8}b^+1xY&Qkw^4)`)Uf*?*mNFI}YLBBM$<DuZ}N
zs^zdw+t&glgWed!iM^#&BAcM3&F>z=LVy!2o+nQvDWy?O41h(A48BaGB0wdcvaf_o
z5LWAX+K*ANazFvZr%z1t6k@259$Cp(C{GKdEkbF7{^HRyEjzRn+mkEsl&{zbWq9X!
z1=QM=GSMn@vr3Q&@5k)xSz470TL3!7^!F-p`mWN<y!5^M$iSd5StHSMI39x4h0rts
zAlXV1i1^f_H+^M6{#mIu;@5i}kO7{R(*sd}lJbze%hyQYr>z9azVcLFFj;4rM^Bvw
z<cCxXoQef#RscvMz1+^(GwY!}xtiL;%c(uJ!abESdWI33Ax+hTo<;3}HJcmOY7+v=
z=1v8Iddb154UaitmF_A8i8naYDr?9j`QwMJ+TMh5#Z}SQ{RuPn)o!9J#iMGkDg<+B
zk=WnX5)A}RP)Y*SO>tl48+C{3MLKv>7uWU05dM^m1%|Y|tu@CyeT8zCBSm=)<JDb?
z{zD6i{iIp2L(5&e)C+AL&z&ae&6G*XNt=dg(__C$d%dx>?gmML-htOxx6aT5_$u|d
zBG_tQF;{d8aSew2(Nvy{FPx1=)pY0qCX(@kQkXLj=n)|5%WWxFMVL+s6&F;m$R$(+
z<&Ojf=D7w!;28QOsyvDj5C9=yRq_rl?R5F86Go#n*I`5~HzV6RzigZJx^*#zo;U6Q
zoXv9kw6Fv@rHdlkD7Xq9$l;}0m^G9yh%61O3t=k|1bOJ|y*uAQXoW$yC$@+zwYIrz
z#U=nG6$*6{A(n(eW$S?Cr>06Pxtdnlzo&PGP)<yy87MxX(<6s?;z3CarhfMkN41|Y
zLSf1pI5lVJD_~Hz(Xv^IY=5Ga+7Fle_ML^;9%+WQ$^2jW_n)yR?|sB(MjBSbOGIyj
z>U&9mO!OjQl!0AZ>Hwg&Y_qjSMOY_CMAnJgdM<FBBA{h(xZ<CX7dmn)mn$&;<aaH6
z=0}#22Lc$9aB^n4ewd>>B+H6PJ;*4c47t6QoU0=v5NcZlK*`|&6|B@jSo{tp0ICoO
zts)%MK$Ba5oT^YI+iCiYTR(QUJ@Du;dug5W+HP8~bk3q9NAVcFii^U#1MotZQV4yL
zZI9%u<fl&vP}0j+?LlgDGm}=E9I@G%DR2O^MkeqE032xa`de<00#4xh5($3XA>?*S
zS^*tkV7<q`V}+wn5&zi&sanb?)pNj}yRO__ZB9sXNb!(9H93!qD=fCa1|dvnX*{7-
z)+&>S6qX7FyQ%ehAtC_YY7s-dL}#;aGdJCA2R`zv_RTMRi$?4O!fqOoWG&oWuU|wC
z*-56vxN?J@K2Oe0lHqFBsMTy_jz)|5N}$3f2`|ALUiW7N(A#R-*6NaNHCJtZ?_SG&
z0S}N6t~`XqF=530j>!UBzyBLnIsP0jPMJdpbys>DTb0@kK71A@_YVT%9qam{C-qOu
z&jH5)aT^Fo$t@8;%2YfSBnAFnWL&4$$E9&=JY<b{Z*6vf;_!ksYoj&C6{BM#HjYK}
zb-kBJS4tHU6xNzUySZg6oxTmWR;|;<3wxXJkS(!85U*?5ZEt>y&EK%!{`5co7mQN1
z-k@W-FaPiu;h(vVJU|K0-}T{KwMv_M|JSVe$`OmHpQ-?<+AaP$Y6W=cm06X`gg8^;
zOjXw81$cv6VbyL4l850$36VI*;Eou#$QMoSS?iJ`q)i%|%2*xTF%1@|+9-lY<(hkB
zo3I*mx2#Rg*J5be=>cx~*6MYb3zo!}vIz?GjukMZ1<_7t6U7gJ#a(NlxQ~AF6ZY5l
z-D@vC_ETFt-oyh?fB=Z{9%3PYgu+cpJbJFa^}VlJ@%2|Y7AR4r;5x1s(Ya{F$3sU>
zKvhtt4wMk_0KrZ;AVf%tSXcnXABvJ?$LDugIyGZ!`MO!7;g7ija)IhXcVo*sttKA8
zw@s4b9wjfPa8RbFPYD1gihE#_3=3@awyfV8_<PBEMlsmRDq6S6KKDmztNGJf<K(uQ
z9eC>j+j(G~a=2l+&p&cB)iRf2U*%9(Adg!WDPuUMwg3DNEx*13VMJ94qm<>8rorX@
zFhtFeR}|$Awj?HHa;?v(Q}pV>ei5bVIX!!}wSBiO)Tgbveu+fcCyqgw2#_ZS)Z@U`
z07mVtO}veXQDB$*1{Co=z^N~HqV>><v-QCao7{)O_2!7Ywx~{#WUb^VqX~;ufrZ%4
zAHu37j`bn|-aE5*0uOlfI9@5-PHAJJ-sY+`PrhImj~%sh$Bx;pqj{U2sz4Yw!*@M<
zlem<`1}J1eA}moN6DAdLe{GHHTDDG5P(obhVZhh=P+A@ZCx9h|!&&yb#OtM%mTe+*
z0&GpTBD+BMsPfv|Bvw>75T@6~vk*{@sud!1?GIm7wmFRlf>`|`Va!H=NqtU%^Fo%Y
zx*A7LMv&CD1is`^-#C0R-MsLstsMP{Ex!1?CCxUj(z8vF-pSJ&_VIV_^A;57;jvf(
z5Xmr-Q;1Aj7vQm8+Y?-|1W~y@rAf=&2{EY<6H?SlkRG5EF^5)=2UfDASaN_>&cloh
z0drq};Iy5*ut9#qa?lhBg^v^rc=MHMy#k}Ms~0t}LsrHmxUd6VWd_7h7<s_)1lPC~
z7+$j3I}X?j_dQJ>;ry@v%0Eb3%T3}?ucL+5LHL-AT>)5Jd(#1HoH}mhlvtEC69FRd
zH!*-b)}xMB0xXPmyFzeKNcwF`A>}ee9V&YyLfsotzZFFhnK_n3FqMbV5F{wT1wbc3
z7H#T^UG~2ZownDWJLTvtMB5317o~7+lfsp#eI?1~Q{KtEqAvwARfzC{V-!|T*s$$A
zh&)M;%+gAQM7oLxpy2=ZyT6+TBXz5fPPogoVC+pd9Iz`7?6b<mq&@Uo{~V!+U2wN>
zzp%S9zoJ=hUX%x0gd`Y%>#&LsL{=gx^pus=o<g~-%#RCryQ|0JFy4o8uM9L<1w8~9
zT9L4}?>u(aS9pZ;^^ptfqqoc?Y#TvkX<rFOa9?A@7_rbQlk1%TxhSXnUOhe^0MrH~
zN%a-MT-o;CPDi`kvhe4A=>rEBn_X*OTC($t7j5bEEB4Zp&mtV4fiPG<0tQg4D|aGJ
zg%9+|S#?}-tVSdb(X|q(h^Kl(M7@B@o#Ic7$QD}P3w?1@0HS~|t6XYDSiY%c^FFzw
zM)^FwfEVG~B^01_LxCzKG$l6}unIz}(&)u(VSpO+^_)=(Qyx(3F8D1&<wz1K6-&25
z4t`Rbe9JsQXwx<>H7(pTGIFrJbkVwNtCWl_M)f_`Pn<k%jqy=SUVOn_2Sn|H;I!V{
zD4QzA)v#!>L~vIvv9i8aW3@b`x6T3>)5}vzj#|i!gplAw@A)csWb>c{T)U_;g4di{
z>{^HUJgXzP`vGpPX9F@WJt1@BiG*xX!s7{kfr(TRG+s}F!c-4rNXTWv4_fjDy)-dc
zZQH`ag3a!pv+(b}=NAu_%5}6!8%x&RTu*FmanmlovTi4TOcmHi@6{#IhzK()e0>+j
z1dZJ?2rdRx7G;qT;nb8U4)cP!rL1yr=-F5pqe%P<D`+)3PspfER-mFLJGDeb7`nLP
z#i2Pjrzpq&5UwcfK+@E^16tD%7*aW=@Osi!vW)Q%J-W+O2x0c}@E5mx)}pE;B<yhC
z&T9`YomsGRFE85S=>=OnbJ135F`G-EuJyM4!tJ|^5(_gBMo52alZj#iEX4BC_ki9&
zOJZ8L6T=~~E=_sTBSm|%Vh+#HQX*gF1ih+8PV`FQDGM4QbDUi0`Vy8NY02ks4_q2=
zKzke!Hd<&S0iy`L!yut#0Ews<FBNW=*A&X!poU9;Ore)~3nOL=Hoa%oa{uS>*>pgL
zh7}S8E}cNUwP`(2Qg7v))gJi2{w!SpN&gW_`CkhkF)o#&eP;9WhrUTfJr1#K2gSg!
zasxG$(o?2NGPY<|#`>s-&XVy(y%#7zMe@Un{*ndEKls8L6^5}eE1_a3-JQXu!I`K;
z5`x1F!U24Ub1%u9(f>V`$N5X(jF;i`7Mg2Mp$HG%5wqDF<}KI+dTy?-+xpt3b+$lY
zJs2!ZSZVGm+qQcjlWBdgNO@!ew5TUOYDuFA=3>+l4`^z7WPS-mO=p;dcMrfTlzjwP
z%FI)Ouwq{BDdt-(Fb7IkF3~&00~GB4mCh@L`U|dUDUR?cAc*n-MH8%$^Y;?;pR)8?
ze2!#XLCA_iOWrO|Aw1+AUN)7($VIi?V^-Wb;`yn(8d>>5*Zv>Rg@Y<8a-i)10000<
KMNUMnLSTXg;ov#|

literal 0
HcmV?d00001

diff --git a/test/test_files/mask_basic.png b/test/test_files/mask_basic.png
new file mode 100644
index 0000000000000000000000000000000000000000..0c2e9a6899e5c0381ce7c7364b31d684464ab423
GIT binary patch
literal 362
zcmeAS@N?(olHy`uVBq!ia0vp^4j|0I1SD0tpLGH$#^NA%Cx&(BWL^R}Ea{HEjtmSN
z`?>!lvI6;>1s;*b3=DjSL74G){)!Z!;D1jS$B+!?x96kznhkhd0>ACglv+8xYZYf@
zg7GSo=i*7f^M21SZkWY<MP|V>22=JGmJVkbv-kp#gwEW^`mya*CxhPGhwnXiGR`q;
zo_<(}c}Ira`{OCB1*f%^@py1P=&GJ^RapMOKMtQgA~FY_S)9o&KEP?c*l_PbMs=N@
zV_(`fFl>98;NHv3x8~HrJ}GvYh^fu8rhFC~wAih)4l!|Ub4{*lX7-Da=zYmMk&)H<
zvBCDUjP^HV8at;yV&?jG|L~f5QyQAI!wgd$!W<6jey*@rGhFask>nnkY0@8#W!nC+
rec9Q-ulM)Z_RxN|lmdu{w!YU_dMY9qnsK!N7!(Yiu6{1-oD!M<m$QmU

literal 0
HcmV?d00001

diff --git a/test/txt2img_test.py b/test/txt2img_test.py
index 9484fd99..ad27686a 100644
--- a/test/txt2img_test.py
+++ b/test/txt2img_test.py
@@ -1,77 +1,69 @@
 import unittest
 import requests
-import time
-
-url_txt2img = "http://localhost:7860/sdapi/v1/txt2img"
-simple_txt2img = {
-    "enable_hr": False,
-    "denoising_strength": 0,
-    "firstphase_width": 0,
-    "firstphase_height": 0,
-    "prompt": "example prompt",
-    "styles": [
-      ""
-    ],
-    "seed": -1,
-    "subseed": -1,
-    "subseed_strength": 0,
-    "seed_resize_from_h": -1,
-    "seed_resize_from_w": -1,
-    "batch_size": 1,
-    "n_iter": 1,
-    "steps": 5,
-    "cfg_scale": 7,
-    "width": 64,
-    "height": 64,
-    "restore_faces": False,
-    "tiling": False,
-    "negative_prompt": "",
-    "eta": 0,
-    "s_churn": 0,
-    "s_tmax": 0,
-    "s_tmin": 0,
-    "s_noise": 1,
-    "sampler_index": "Euler a"
-}
 
 class TestTxt2ImgWorking(unittest.TestCase):
+    def setUp(self):
+        self.url_txt2img = "http://localhost:7860/sdapi/v1/txt2img"
+        self.simple_txt2img = {
+            "enable_hr": False,
+            "denoising_strength": 0,
+            "firstphase_width": 0,
+            "firstphase_height": 0,
+            "prompt": "example prompt",
+            "styles": [
+            ""
+            ],
+            "seed": -1,
+            "subseed": -1,
+            "subseed_strength": 0,
+            "seed_resize_from_h": -1,
+            "seed_resize_from_w": -1,
+            "batch_size": 1,
+            "n_iter": 1,
+            "steps": 3,
+            "cfg_scale": 7,
+            "width": 64,
+            "height": 64,
+            "restore_faces": False,
+            "tiling": False,
+            "negative_prompt": "",
+            "eta": 0,
+            "s_churn": 0,
+            "s_tmax": 0,
+            "s_tmin": 0,
+            "s_noise": 1,
+            "sampler_index": "Euler a"
+        }
     def test_txt2img_simple_performed(self):
-        self.assertEqual(requests.post(url_txt2img, json=simple_txt2img).status_code, 200)
+        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
 
     def test_txt2img_with_negative_prompt_performed(self):
-        params = simple_txt2img.copy()
-        params["negative_prompt"] = "example negative prompt"
-        self.assertEqual(requests.post(url_txt2img, json=params).status_code, 200)
+        self.simple_txt2img["negative_prompt"] = "example negative prompt"
+        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
 
     def test_txt2img_not_square_image_performed(self):
-        params = simple_txt2img.copy()
-        params["height"] = 128
-        self.assertEqual(requests.post(url_txt2img, json=params).status_code, 200)
+        self.simple_txt2img["height"] = 128
+        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
 
     def test_txt2img_with_hrfix_performed(self):
-        params = simple_txt2img.copy()
-        params["enable_hr"] = True
-        self.assertEqual(requests.post(url_txt2img, json=params).status_code, 200)
+        self.simple_txt2img["enable_hr"] = True
+        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
 
     def test_txt2img_with_restore_faces_performed(self):
-        params = simple_txt2img.copy()
-        params["restore_faces"] = True
-        self.assertEqual(requests.post(url_txt2img, json=params).status_code, 200)
+        self.simple_txt2img["restore_faces"] = True
+        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
 
     def test_txt2img_with_tiling_faces_performed(self):
-        params = simple_txt2img.copy()
-        params["tiling"] = True
-        self.assertEqual(requests.post(url_txt2img, json=params).status_code, 200)
+        self.simple_txt2img["tiling"] = True
+        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
 
     def test_txt2img_with_vanilla_sampler_performed(self):
-        params = simple_txt2img.copy()
-        params["sampler_index"] = "PLMS"
-        self.assertEqual(requests.post(url_txt2img, json=params).status_code, 200)
+        self.simple_txt2img["sampler_index"] = "PLMS"
+        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
 
     def test_txt2img_multiple_batches_performed(self):
-        params = simple_txt2img.copy()
-        params["n_iter"] = 2
-        self.assertEqual(requests.post(url_txt2img, json=params).status_code, 200)
+        self.simple_txt2img["n_iter"] = 2
+        self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200)
 
 class TestTxt2ImgCorrectness(unittest.TestCase):
     pass