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 000000000..e69de29bb diff --git a/test/img2img_test.py b/test/img2img_test.py new file mode 100644 index 000000000..d8ed309de --- /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 zcmeAS@N?(olHy`uVBq!ia0y~yU~m9o4mJh`hEk44ofy`glX=O&z`&C3 z=XQF4cIL@7qfkDmQ)5S5QBF=X<=a$&?()-_SpPxHCTm9_$O*S6eJ~G^pnB%}4 zD{y8?YS?76SY5uQxlUcP`4UxqS6<5s_nv*JW$GG%XJvv;oJ&mDg!zvg+s3nPW3SHb zNvhu4Qh49LeChjrZ~1fMFW>J!@A_hQw|d$8n(BMM_wO&B^7~YJoNutV?sVh7v+8uu zMjkMG_~Mg#g2SPuIrnwaR$pwK|8e)9$;tkopUf~_-Z-l-zpwFTO}%x!)}orCy3O~W z-fr>jGqjt|fAZT`|M@v7WeY6lOq+jx-=fK??|F5< zrH1ZFPENm$2<24h2*-$-2|L|>cvk9IKgTl5h=~PrxFrO5pYQSUpEt>{@jy;;9#`8t z;aA6BZhR=sV$|UFx~$?;w$&wZLyN+juRa|SnSQdk|C_1bbtj&sf-{9~Jl^v|^2b}} zWk1AsnC?_;-PLkYwKg{PT+5bKottKxoU<3WB9Pt^oOx{e{hG+nR{#I+UY^}o&&m9! zKl$v!owZT1X&c&37MH{oyK=mE)4{PhmQTL-;vL1VDIa!47I!Urc6qOzh+dQk~VrFfu#kQoN`cr~d*J`-fch)L}-L2huj5q1f z^X4|Ecj;kivo)1wM{`Zj+LT+S$K9CYbEIPV>G0z-izH5`Y%0>?pL8L^G44${-^-m3 zte!-~+ce1Te~{AFT#%^u>Q)2i5?^2A1TjDB#P4$!nyYXuzsusi)bU`^(nX?Nma{cV za<@#0S~}(UTFd#|i9y8~e*!G(m@avOuRm|-G>v1jmbJ^dW4-?qOK1c4tQ+|cx@4Rx9k-MwpE$~JRV#0 z_54V4Mnzmh08g&?EzwCIpM7ziQykQAEPK}JL%Vu*_3wXZugJC|Cc+ETWzSvz8MqLx-)j9T6o@$pLcw!()^rsf|OYHak9YdgHC zR6#tU;UZo)Ue5fT z^ZEjR?&8^-wj5`nW=G1KXJaZ>&Hat zv@n%*ALB3f{5f_cVAPi8x{${)8AR}WH~EVJwI=?(sqb4%k5WY+T> z^(!+>TrgwOf;cUUbHyUOwd$vZIi6KZ9$p=IRWCM6@O9%^|LM7T%U*qy`xts{YWH_G z=2J;Ik0a>gJk@zwVw``;#WGjR}TbC)!!} zN8b}W5ZRPe(qEUNDt>YMoRVM8x6TDWYn6KZ@lCW|VO(&EaO&(m>k6IA*89DRn>#tE zW=6bwq1_So)Z01kO)r@ZS{YPI{g%&}A@MvdWWMwk-E-nIG!))#ENfJpp*@{nH-7Cc zx1cTQ;auNxjMMe?za}ajj9wL{GgU&mNdBU3n~8Zmr@D8b>)M9bo!cIB?U-hsaX)eU zjtN0V(|B0kausE6NzGZNlsG9T@2vJ>EuSSUlJ+(C*WWF^(0iUU(?#8M;{2koMfc2o zj`nr@ig^3{q`>p|jZdx$i-~k;=q~(s*d$VC8m~vm<=tv;kM-ZnJuM!l6B)nFTi-tS z^!oW5{ylT$-my1eWu>a-w9^_atX=_X(?f5sHPwFRotLY6#7D+Zjd7`f{mOa?WnmBZ zbqc&sc4@WWKfY1P+gRgH$R*o1`wCu&a0Z8`X68+=v6ACj<$2WM+sQqS>$KN~oLX2q zukJ>Z`n$~wI$S@ewlv91`fAO5da{J~rTyZP>xw)(jnoB8c>#OKxVskU;+ zL}u1psoHts)z!l6H#gFjcFLL=nWqLW&8h!*zV2|{)6el+e|_;YaOD?M=1;R+9_|09 zL;J6Xgjt@$zF$+tIox*Fe9`Lq>=WYDEvx$dj(9Aim&YuF-!Wy%e=C}u=PZkGUH;`O z=T8yskni$$^DheXN9CsUoRnOAvY`IzCL4_noA+hb|GTp3Z?}K_oqKi9t8;P}r@F`4 zCHmj{-;gV7*O@&%CnxEOnc4N3=J|nFwoZweB=-55UU<%;dmE0e%MH6*yS4ZIyzP%p z^cYV(aaw8LsXU?RLj`ejV&A!aow!?j`-3?Sb&~qh%1=)`+IjvmuUBhP`@)N{jFzev zY^m=a@3&iMCKU8^b~{w8HT4NAIS^T4$F%hAo> zy5;vA{r7PHPu|@>zeVTP7N4&B^oXjKjj@u1ay=OQz!jvf2< zX4`}bk~3#)TA;RGB|wEG^wh^51CBl346;H@>Q6bWj<)hxcvSpNGjKe0!F}F@r_8Ze z^Eb*xz20ZP`On3}VXH(pI!RtvIk`f}VZyRIjaQb2otLYqI#~ARex~<&-CL(F?b`aw zw)!}G-MN1s)$8v*OZatXX=(M#6M@$MZYaLqSH?CgYsHS)r-Tv$Ll-;RZ`yM~`1z%e zU995QHyss{m=YT7wCHxOaASzLR!BqOv1e&(tv261lxv++Qg!XnMo;!Ep<~sv^(QhL zu}{zTFmOqCO08jco+EJKgP~q!*fNs^0jHy%?^_>P@TNo5Eh}zU(t({e?wqbWmBkrb zU$X80`r&m155HjIx~&lFrvt3*!9S}q%`DX^iZXoj*chhTzII>*9E)omivgBnxTm2FSn80IKYR>G6v zcg}ECc&RjJIoHve2k)x+wCq$Ci71Gs+zeGWzIx*8RLxjNyJ_-eTn1O_e{KkTye5Y6 zEvM9?y)}JR{(ONt&o+cfKAYZ|8Mz@~S*6m!t6HqHmR?By|3O&)=HK}J;?wVMFuz~6 zSawI=e$#KU*+*w`MqAe(+a=AQyF*A%;!CnLPgPj!%9xJd$)!^-=ymx7l|oMq1a|G&yNY`GbIudF+^^z2l#|Hq=&cuK|B{ay2Qw(V-cGOq)lE-B5s`f9nx zx@=`-(J2l8GWR`Uto$VKWNm-bua@Y>Co%yicgcizDX@BNeIvT>z4Ckujrm{Wq<>Cf z(XS5?D~@yi@u~mkj*0P_3~jfc$8WjWo7z45^fje#R~e_RayD4nw~njhAggms@QjCV zoTS&EoLpPmGJUztp%w>I)oT+2i#WWhejQx?hW-8z?b$c%63zF%RJ&dNb?Uaa$6l`u zHi@?Ry6b6GtpAl0d0WE5Bv&h_h_#-3ppKuW`l^&b56;vJVrxCcRkHQM58+YZRx=gj>w_D=zTW z{&$>x-#*`b-$%9C`-|?S<@(3ww&sC(04~%-kS5+&N)n~uL|T9YePF?5BfZ0s%#dyVDoiN?T5Sh z8xNniNmjT29(sG{TF2eqn@%l$I^#!o0F$ENv2y89pu$dWp(l<8MC#}z5 zt$|}-c41dOQ-+s@L$ODs?G4+qZJDoEtL#`<^dwPe)`73Bw>bO-BVJxgn!kHPT7LC) z>9_HRI$s?pWBHS?NcsvJ zcW2c0iB%^*g*4s?yFAS>Z28qq|FSOY=`Nnr+gIMPmr?G6tBu3VD8>bg)U&;rZ)tIQ z8CH7T{4{g=+iR{aXHZB?0)=MTQtC~V!5vi~pp{7vuwJ>9#lq;Hdt{6f){GNt>Y^qJScUU1Y? zdMT4jNJG&6RPAo@%>n|7I~*7z;=`U_IH+g1N@d%wl7s8!zqx(&_xGFf_daXp#Vn`` zee>wE_}h2Z->v?A+jmb|?XL6MFsDZ$Z~S$GUK#F)`(3eP%ESrUPa;=oEzzB#_+iqG zi`=?`Z1cKM{r>LzZGX$}q}~7f_`hrF z!QXD#Ve5ONJSs!%g8jHS8f7k69Y~m_e6#P0?)TIQ3lui2(O>xVUzD$4F{99$b#a@1 z-s#%=_Iduc^M6kNPvBlZk1-+S&6|s=vlnFke7gO7B=;Bf)vQ?`Hoj1BQaQk7Qmdz` zUK#A8du_8O<17}nQ0Yr6k{^h?Sk;zTJ7rygMGYT+_LMc-(%XA;XK%jvan;fp70hQoR0V7_zWwa&o4&gS;YYJ~nCC@LHQc`UFQ0W~-qF7K7J1&ex1Pk@&8z%x zz2?iZO6#hssa(zK2S1;=TAlm<_We}znip%Q$?M+SH~V|k)Fakm`nOb$3qMYsJx}NM zm6Wq?mf;6DqSn0B*_!+Es~XpX$E(-g`uv4|gM;$w^|BlL>z?TSel~p{n^E;0x9z?+ z4|#WoDR*A!P>p;SCZv$XR`IY%#r4OrD;&K+QkOmn6li?oUTXBzG~Hk4SXth0IqS&G zh<^v9-<~-tYCYXASo6xWN3xw!_Znjh&*n+rt!ov{4i>w%QYutxeX)41n(pmCA*RWq zO~%5>3>$f^znj}{-5`4N)N1h+5v67Abr_1l9wZE%AA@~2<_uQKEJ#+20C8Qn`Y24x(^Wz$`K^ync z!>(r;m>9qA_mT;pvD}~YGFPCk!n&Eo3~}e1-{tLkYqh)K!uk5&fwv8VwobeoevoBp zw`KDZ_eZaHtQLF8UY0w1ON^}Q&{ zGq2ewii-vY%r8uxw54H2RfEunJszPajQ3qyA^2&>M7c&@!98pD{$eVu<^344@$vKN zIq!4LEBfypZtn@>JvOx1N&?6N=V=y2(xvUcj-1{ni| zyK$|j&%L>oR~-I!(@WW4Gb`h4%~QgEZSGCZDil4sbjtcz>$Ejiw$0l+x377+cJ+FN z?-h|p(;OXpMGi~r9x!xW$2|KTO zTEOa}!AZ@VyticSmj^`!TswTWduyAok8iY7pPXwlLGT1zcsA9i%?WZH8q ze4}AU)Nih?kmD!T#mG+&zSjTw+M8!@Udr3%&j0=X|J8qw{Oz;V>;JUq?yqTk!I68v zug&+;zBY~Q@3ZeEXe-!c$sL;bz#*HNp}%W?K%C$q)((!ZEBAG4C{Hao@PwCJE!{Jv zvVniCme}j0(#j|%rjm%Z@ROfhXR)qjnwnM~$^37-WBqN3g|19;^QE@m?3{JB;bW-O z)g_`QeOafhW6@OA2=R)DmbZN^AQ$L)&7G@RH_wGH_RZ0SvhmtCkKOpX?Y*u1)}Qe| zHpo`~(A@pwQ0DHm^E1w#58Ly+|Kx-6JB8Xl9F_q^EV-#iWnTD4F!b^}wAdbKyK&*Q zv7ugcfPCL3QKi-_-DI}u@{CKTax98?A*gzy=#s;9J%4Tqp(QPo%KXD~e@htlt8dFx zPuI4b6nVHRJNt-fuc4y2V(6j2wG5N08d@UG2XfuLqTzhNC)J+0YL4)e(z)Dxf?QQI z6z8mYUUPM$vrs+d{c?d zmR(Z}KSs7*WH7j_AiuZxe@ob|H?Q8BR)4;?D|UPNm9hyhpSv+SX($H9E859?E;!>B zAKvoFDZ}^wmx>k z*)lsg!jYN(auxHxw<&Bbm)3^&-1s6fkM+phd3iTXKR+&eJ1=ne*Ik^|ufGPD-QJp9 zc6eUf+nauJN+nqr3i&U%i=8RE+L5wHyX^GsZP7pe2rgK5s&Agi<)aIKvY5+6+V#ss zNTjwzt)D(s*(<8gFyZeV2K%?$&PGK??pYe^_TX|h&gv2H=eK{-Fk;1-Gs=s&I_8o%Hg}2Mt z9cs|?InFR$;n<=pKb_VxF{jk8%T_h>?EDyY_tuuVg7Yp^e2#uQ_3AX+j|ceQRM-Cr z{r&8{?3Voh&-QOST>r&#b%^x78H%%}x{0zrFy&C$yutA#hkO0R$w{2yi;iE{-SRIr!`aYzhBko!f6+ z#60)r9TTRY+3YFTI*P(?*3T`~k6rRuX|b23R&#~hs;3+LZeM3oX}%|MxO4B)P^H-m zmYjVrci5>xSZAKPzl!G4Kwq8)*Iu4|K8uwcROfA4-{LJ0_PX*`f8d592gW1I{e3=* zB(0br7kM&V!=HPJu2W|7zeg91b*;CE+@7rcVaoa&9H|x0D_F1Gnrd`Wt7Nm+1kU%z z-@J~Dn#vQ+V0G-vcdf!->n@$y^JVYv69*O;L@Hk_vXE=K+Iqo+QQ@YSP{)rei_L3O z9sfnHG<$TtT6O~Cys4L&0~;oNny}~eb&m(j*Rq`2_bDXKAToWXyZ>CaC+#8JbGleF z!WcZ1_TD*n)A;p5+nuv+9X`F%cIzSAsfJZ;g0mHUuQtzfecNa{Tl!j-Fw?1LeymMZ zCl9XV?YB&v`c8|b)7;{+R`v;%pO4$u-n!RPcr^6l^}n3P->&8@(mA>~YrPgrapuGu z_pb@@viJsgF`P`#r1oZru8g&9m`dlC+ zI&N3r)}ga?+s>S$Y78o(T%|&a(^?!;UpG8izRXK|=d7@s`=A8#pDNlCm&Of1#ZaukcUp>+GW!r5(yfCcJocrlk zUetz?bBmPSKT7QF>OQoa`GeE+AdcEex&2R8KUwb25pd~r_*=)UV&lnSvt>5koT*vB zu*FU70~>=(#+P5wc@voY9{d#Ith=N5q|7UdQ8poA@A3V~t&9By|KDU<8FfnIYVm=2 zyVu=`&Gqvr^7uHx&eXEMTU%$_0X9XSj7*z_q1Ut~w~I{UV%@#+Yn7rB!@Vp)=e;iJ zoAZs$@8+=EeOmqPtbR?m+Z&ccc~TQobnnEhk9Xm_>ACF4j-uvycK+x5COyo$w&H3| zN}pTzPp<`s)HwAIO2>UqXSr}Y>(;eblX6YNbL-y6ovvl%pCGW#rsarX;*r07Cz+!6 z|CbK4{3oO5H`AG;G`vbWdg?7+yBaa|@BB=UWMBVUr2f_a^m)ePjD^x19w&7L9ZrSV zh^esti%W@Fe*O5;!fczA-irL(vW7<=noP47n;sWkoUClt=lZ#H?{trFFHs?pYqOkp z?@ZO6yN`EU*xJj-xT|M>n4r7SSzvwk&D{LD?z8td-2eB*{-1bH(AIkkl6UI={~+;j z{?0xd2hQ{NzdnrF>NUIS|MfLLP6>9eoWF`;6mv{KzNt+7s4_ z=BUK0d7P8lAG&Z(v^f5tPh$HCao=~5uO}%l@s*l=Dnj;-A+OkqTN+L7PiIx9%-$eX zvVD#3+3QIew_Y;l{krz5bO!gdBhxo=a((F&HCV!W(Jx*2D1XDz*z5&6`s+Rkf4lzu zZnnPNx8P`-dqfvFw5eLR@9W<6Q6ImCTZm}{E`4CNPWRr_dlXh6Qwx6{7?w+;! z=jEu!PqxjHt$59NyE0*3(T>EO=k1G3t{P2Gihmjqq0k>Ks+nxDZ0r9&{JQ)1F(tZ$ zaxR|objq@WPyZRXoShIZ@|16zidt&X+Vd-dzq~qBC%E*brET@Lt+m^AuPk4?v#!(S zrh1>_42}uv>{lKixW?f2cCGK8-p;^;>A%@TKH42vsb9Y6>xHh(Uk~!lzSDl_>$8qM z+gA8ZOj-LxV=-rUXk+V*bMl8)srG$UV!UU)xtxi=;)_L`<{iJxgw;d~>vMZwM{V;92ylvdmU!U$ zuYf2yR`mwe*29tCb{~1Ybn3KRr4ptb-)%~}9lo#lbnCWCScrwdvg=1T3k3A75>+aB z%O4-Q&9axhDdMkE;u*IF&sx96&rZ`;?UHi$bDMBw8_T{Yf%h$H4DaswuxhQ$-%j0q zKlZ4eHCuOSv+sm^CofBX`&j<&vzW+>hZ`f91Wp=E3hwZU5`VBU?zV8uMwY;xc1t}^ z-{CY|DIgql#cP7x0ujUZunFv8o0dcrHH4l!+{43I9AE!n%iKL19P17jT@c}H?HA#x z_&hapFVo^ptHn(>IENim-LZi=HkIv2fJEhik1W@Bh)Tq5x@paKgZsO!uiV2w>;Ip! z_lQ32`zT=j;?`@9KW6mAGDgHyUuMjnW0m)GO2iaXVMe77=7tS6*B1AMGwJk{?sh5& z@c5PMrTEmyMOLd?#p$ke;C@k8!I_rRTnwhRUldT$IKb+l?kJY~sw?)}e*U@JuFtc6 z`a;*`HfNLg(t^Cnho-KmRXzDU_}CwqSd~{tdg0_AiRe zkDJ0C-J&AV@OncRze|{s=9ei`ug-JYx#mnicg612#~&NqYcE{SV-m?~%OVjqal^Lv zHf7hsD_TB(oyh%q@%A$+Ins`)nem)%1@Xy8BsN~qn*4e$!%mImf8uQIQkBb)UPoFqmOcgZ}E6EZ2yAD^ow) z?Og2k(`Ijj_`_SOA`42L8?{tTRLvzea5=5M^x|kvVuRF|uX6&qg5?5U>;2Gme(E8l zDeAr5iLWkbcFk!Uhm6Z+dZxN}jE#ym>@B&tD|_{n9$mMuw%7dU*yQzo4D^nxIT*|C z^0w*0mV0+JW@{b0x*=nuOF_8q#!dUelDFC{-F}Wq;_~TWufC`gJuMR+xV-I{rwYxF?Ik|hgQYQ*LyfQ_D?{m}6rE7``Ufhh3SnTnKTRX$l z%KyX)->?OX)U3C>B?^AsLKKpL6*xtR9y<))bmlm zcGu|;j>K=DK8SCxzP4zUipraU`{%Ox_2x!yDAw1%_2?yQv`z8FuuQ|*-V4@!-S+p; z+uc1v+{OZK9`)IiJCm4nj(ocGz2TmNcT&`L;lqj>^?WNYDZ5s0O5!_zAxJ6yDa+nf zZhl|99%^z+DHhL(dGSkVnq20|=LP&RQuz+mGJdkf7vo+H;ASnOooU3v__W#p~ zW$^qj==C$6Ti#G_>Z01T&!5`&dAiv0ZOmamdSF8lv+2}J!OOgiwq=AJ`?aF|+@z1L zd&M;zTt6M|`ZtO7jQx*8Qlf1e4YU^~Y2KZCaOuIBD_n${4)Yvhtm!D$KRM^(f7@xt ze|kUYnf~yL*sN^-g&h2M1vlT6ex`ear7CaT`rE&5Ei2vTkYFpJ;}enjPV4rzYWc~T zwd*}9Q<%1_P`AtST+Hz->53qKMpjiy9jA+L`iz!x{sY=mYFQ#qoe6H*kQDb$^I5d6 z0lQbY-bL#p%p!08Pd%sI`ipyAre@y<4TX+RjIP3KCVc)b^kx30FHtppDKU;Gx}3c4 z>G><#Jduqvn|k(;x;VRtMa{am?6D)Oh(^?&l`Tbs4CiWaU|>mi z^mSxl*x1kgCy|wbfq}EYBeIx*f$uN~Gak=hk;1^h@ZZzLF(kwJ?fGcFW&<9Vz;F9A zrB+VwTE$tJV7$uYxp>m=yx;SS8)h+Iky-GJ!IXW4rNddqEWQ9Fp))tKer$Wy$)NZ4 z;d{@WjC0JIrymw#-jN~q{&)&&!D+2!JRY16x~gYf6_!8nkHcq=h|Ga!7H4vc4{%y9 zHr#uVQC+9!*q62q4BMV2xc4&itvPkDPl{b8VrsLjDWAm#Eq3dyLrh%TT$8Jsnf)Rp zdSCKRWMs8|Y_R<-qx}t;#?I-Fn7O{)KfGq%l!hklFvC=bFo%P>pDXOu3>Q3DB)Las zn)HWbnYMpyUv@U|>-{~pJ+z-Kr2yigt?%`fo{9*DW?U^`U|?YIboFyt=akR{0GG3h ANB{r; literal 0 HcmV?d00001 diff --git a/test/txt2img_test.py b/test/txt2img_test.py index 9484fd991..ad27686a9 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