From 0c7b6eba3191eeccb43cf49b8033f29b439dca28 Mon Sep 17 00:00:00 2001 From: Susana Guzman Date: Mon, 5 Oct 2020 15:13:23 +0200 Subject: [PATCH] test: add tests for plot in end, plot in middle, plot twice --- tests/unit/flow/flow_original_end.jpg | Bin 0 -> 7497 bytes tests/unit/flow/flow_original_middle.jpg | Bin 0 -> 3758 bytes tests/unit/flow/test_flow_visualization.py | 23 ++++++++++++++++++++- 3 files changed, 22 insertions(+), 1 deletion(-) create mode 100644 tests/unit/flow/flow_original_end.jpg create mode 100644 tests/unit/flow/flow_original_middle.jpg diff --git a/tests/unit/flow/flow_original_end.jpg b/tests/unit/flow/flow_original_end.jpg new file mode 100644 index 0000000000000000000000000000000000000000..5f129101479ef9f94b6820a8a8fec438084a97c2 GIT binary patch literal 7497 zcmb_>2|U!@_y0$-Wyx40`xsl9GK3Pc4KudvF_s5oFpMor6xCB%Ld;+oLUxla+lZ*A zLWJyO?a5A2_U!-h{pIQVJpb?i|9k!ZumAaM@6WyGo_o%{cRuIbGY20J{sh>Oa0DEn zp#cCI>H!=~9Fsz5X*paqHHIS$bpBOw>=ZzyW9$Hc^YZsI)zuWaW^E&KSaw8?&ewg8 z@PBwxm+5b@g-xV>TU9LM*nZ8pme?MyHRAtdr%;oxD_{Jf2`Y#-Fi2ck> zwWvH#s94JN->}oa;UIq>e=1MQA-}Va-yt5Q;>$SCfWy8H=}==97cUDl>hl`)5&*6P zrhqP>NyYzP>DPFAX9IxpIsnkd|0;7L0YJ?I05~=Lt4#a}0GzxJ0F+K2M?c4ZDRZ29 zKZd~oz)ArCuv!BEXBPmRKpiRitL?A3_&?YxLTw{JwbPq=xB{Ml3m^g@058B9fKm|* zI1k7HiU%Wr7C?LK*x^Z~8ak>)>4D^gFCsR zL7ZIN+}tcDc~9|ho#No)<~kHYLrZOQoQ{c(j){wvnU(8*Ob34eAVy#YXr!eP1CD`c zXhAdwjZ_0@09t_daAhA69X-RbM`RniUj(0V?4~792S}He)76hmR zdxs0|kZ1(Vubq5hMbW@N~N zwYj*~o!HasPKifoa%gMYdz-N5mxibN&#aTy3=E9vx&Zpaufe|)%cS7mrr7S#0kF)k z{6X&1C%vLGz0g&l;L%%lQCMtG7)!EyT=V=209B4a8XfwoTJgBec16|)S z@b_3P_8J!;`S|gZ%bhAnCtnG9c!?4dkqXbNIbsM2;ze3&x^=gI>)6e9h()Pmh&3wY zrlJb8y4z}A(OKTsOntVxP=uzvS(xt%Sb-EnYVQfJ6dh?=7vv7uJ6ll5k)7JSe!W4o zgiYU?!#Frms^hJxzK$1veV@!)ON0+)Ne3y}BN5+cbqFM8h+a2G$mr{GeumpGrAKyu( zwwBTQVWE(uanteIpvqR?-;nRVj7IS{&BXSfHb@;0nudxptD4`N(#_AnR!!Zg6+WR-9Njis+%_B|b0og<0!+V{zX^;aC$8&x7aaGc^)I6lam51cj_h z7tl}ypFHxCF!J#n7X}gJA7}R#27?6f;a3H3kYbN-FSIU~ok3rz^iR1fO~#SQUVKum ztWt40McU^G0qPeV9OJ;43-XHkm5+Je2~=7vlzz7Uyj)Oi|5Y5{>1(Vp=GY|duBES$ z#ZTJGx6k2GB|5idjUWR98?BNNl?9&XL~{&HKO4B05wg@u^rno@z3{)k_`%0JQL zz}U9Ew}S(;d`kMX0cL@x4Qg?QGpp!s0z4T;NG)$pf7dmOe2{OO;~(Ems;$c<=Xy)S zU0NlazN~WH$JuqntubHX_3lj*-!#tNsxb>TaPKepz+_Bag-M|ryyw?`e6Ca<^*R6) z@%!MEUs!DW)m!gn>`p25+iUsnE1QbxeIM=b;KMu*fFH2}F6C23xz!E0_h+n`LYf9g z0xyTI$)4!d0uL_nt8vKl#>Lgfyvr$1af)o0NeeGhkqNR1<*-h(0A~l+u!rDR!N|7a zSY{+4tU|8$OL7vX+xEoq)iip$Wg zx^XS^Y}(EDJ2DTSw~mLPn@xN5Wc_8$vArY`h_|>sG6Guzv#4TRb(-94ixoXSiM6E@ z>5#C&FBP`5B@b)$*JY@um|@!38NI|3zAnyQ|8XxecSj|7bYE2EMOgc3`BvDm(?3f@ zCZyFqj0dSlDh2MPD7q5R7Ui)SjqhO0sM%u4v9W7;ETsX2kqcFCSdm(+c~zoPIEiYZ z`{@Rq7W{E3@kfm{5yK6(XY-*-60EtXR-E*YXAaTbwGgIl_&~n)%(vnMa&p;~v5}46 zhH4;PK0}#HO1IPWsNW3EM$^#r`lZ|Kq@iMWgMRCaGN->=h&63?^(8Kq`K)!dH`r-q zRC&&gsQ0W##Y}qvG>`rF)OgWh=aj~mFV;_@)9+!@M2m-p z^Q~26;2K10kYEPFB|EQ8-BjPmP$Bbaj{;IJ8lRExh-%u<#U(pvwJYZC-qer$;J2b!DiZzPHe*K;*t@3bJ)Z2!|P2wm0qy zPBzO<^Ofw&B?%t@^ysrsXxXDP(Kne09roQ*rK=}xT2n3#qr}YpX1r%_6F%l;}c(RU++`{9Nt=VANm6znpb1Oz8zln_a7MLP6dt9<| zTekT4{JNT@_BjcjOoN{18Ox?pfshawwD`?rqMyFV1~^jYePOoF%mj7UkE;@LL_aWYQr;-T9je zG|8@Mg7JGhwL}J|aoNZGdYQg%7p~MG!=lY>&gHe?FQp5>>tkAKjEpfTi?b7kNswyo z#~yM8`mU~q;@jF;XU4@Zw>6^5V3nGimNvqv=xhY>_g%Rd)K>K{!$!6Gs}x$-qYecu zt0)PkF@!{7J|+aj9(xoW$Z}UdzmiD9d*rQ^e})9nUf+798*9-SkA}mzKXo~)teEYi z6y3L2j03flSLTOGpd*T>fAevDeH;8_ys~sOl)gsEY`+MSwQzF|+!bAdxYBt5%$HZZ zFM0fRJkUKrM%>^Y*a6;VDu@(}W3A>BkBM1Fl%TQdRfM7Nr9GIF{n-Dm#J?&`?}#16 zDN*#|7o-y}wM7Q!5y>bl1fsLaccKGlf8KEopfNu*^%Crx$8VQ$$Y;@fJ+eVU!P1tQ zUojG5xOsM!KmHQ)SGW^8w(X*v_@r6RduGs0c(`wW|F@pvQn?oo=dVh99C-ggI`7#I zo1R8qb00iOyv-hs#%Qe^04_oZIngbb_GTRV{08Vz%htzywS^FAd}L4IqPK8r9v_*> zIyCb7SJts@bmJ?lD>};-=RB``lUimw3XNX8GDt~$!gv%afy+;f(-9I=5}eRWKK zWbFee-H!AAVdk{5_P-mR3wN$myBdJzN@^$E~ zcev@g$E(nHM}gf@PnVWvQdUoWMnmlfl&-i#b{EL8iFRZDgL#8NhQS_X*Y3ViB-37Q zhhal1Vxe>@=QA+_;#DvED&mv5XO@Wf5p!#Oi6sH+Zs1_A99ZQ)Bh6zPNOAGL9P!*} zGx-HkhDWm>n;m2fuxB6TFKBy@RN7r6GkauY_=>{aXE&n8Q6y5(Q3T4Qnk0_DV-68V zARuUMoh*?5Ljhe?VhEkt9`t4to6Sa=o003dT2F&(kL5FfPnx=*Fa+%9Ypz>jd0I|L zLP{=t6=99{ckL&$o^>U!ZA$E?y!_f6agP*L|L$D7***JZah`Tw0OvsYd zrZIj+{aMH+ab}tncx1;>Bu&%vq_aGx)IY_XB&gHa@~S4z;bOt4Yeq)M)f|P@>2^!l z;`7Nq9`2-mB6soz0XkuZyvNa68NzEY^ZEG%XMOgV`ct&cI#yGO61OdXrr@9RC=U1k zFe*0}oNuKOs#bYg#@aVqPNKPq^3t}zS@8f!Nm`n7pJ%!RDzblRGsj7Gn{1)xA&h>slf9+!$7z_8m97!{Tf=MD+=%=a4Dt2%Z0h^S+Pe7-AyUy_xdhV$Zpkht zqWMfu6r6D9W9aA{0%tol7`ID`lHXe1A6Rp;7U|EwAL2KtF~9LWQWi%K|IMgnt7Dt8 zVBSquWm1T+^-t7!QM$4E5UjeWAU6Gl2c#nZn4J7*GaH`9DHEFQfsq+@Ch7$@$I6+u z{-Ky~j^t(_&|t2wu4Fd8zMwGnVA#}dvCWgHd;n-qqq7TqVL1hGbWzrg#4i}z!~_#Q zZ(ax=-$JN!5c`l0%ai;_28ZT|1Ipan)DD`IMPA%2Xdss?p8 z4eBmmcMZOf?f&D==Yf`+GbLjaLUAn7Y4k zwAJOd)WoLX%J3vUui{#~2N|pD%_*ly)9I3N*Zcs`&sf&FGh5Ml<>~Fb&Kpk#_L;Wg zs>FC4c|wwiXa52PyQVq?W0<4SyO!7@DJ2TJ9)_ zsZG9I-)?nd-<#+euJ}Rl;ncPz z$!;j{b@>c?>-5@|j_KJf&f7npb1?N9BRY#e2sCe|w4IFy?*$>@EkEKK#o6qkBAM{a%M+BGi_kacLRv=3yc#ts%AD86XY4 z6&W30)!f4^7-?nQ=jlQJz(3-2bsU0~EZb0~7r~T>&$HQcT2QZ#v61@u;|q#_NBkgs zn}&RgFmN_n-aBW1=1*$+3qQokhJ5#R$en(&R40H@BcQN@URz7G;5q3;HcOCiTZJ!? zeK;~3YeUeB`@(9#P^Rw?U|l>Yz19qQ?8}wu>aCL5D4-)bH1c6~nN@(n;e!36l^9(K zxBaVMF}A^@VT-Jk1e-*kEf5InI&kU6ewXvKxsR8O3{-N!$OQ&QVkHU#s(HvJ*B4;? zP{iC~!DqWYdihrn_327XW_;r&S_s=9IMwJE^y8s%WlehL1l!L&MEE0`nWFA-16>K< zqo}>ybDS%6SE3}!^CFMt)f^O*_|T}x1kV>*$Hy$i7V;7tY#3RpdVTW%=TdqfZ^D5e_x`oFPZo>bku02WP>F+f1%UhBAN< z8-c5}yIQ(5bE_6}-=rG-{;FeqzM~g)MvSD{Z}B3TeX?3c8x}Q7=?%};?94Pt<r6S_AT`8xb~*Z?3IV9 z6TctZWDFz63}=1;0e0{~F1WDX_sLD1T%54(v60I?%A zOp9|YzPo};`aQFPs6?L3nOuZeU;F0C`1X2&D?EK_sInqfl*}NMaoM#FLa0-^`a5#O zG98J)<8!z0#Bf@>f64uIJJl?^JG@A{W0U!*{j_r9GjhzeEo&Ua*WL>$HdzgFFq46B z8go|k;R@Tx+DX)naYnfA7aGc)UlIVIbyYC_W8akNb=B1NIE}=YH#fK|^!7bph^^~s z24{V$&}6<*m#7cbx(09wrsBe0xSeb(Ok#s8rC5nSYCBy4la|}F6Jx2p+?v@Pz@NJg z&6%?Lj2O0TC2tTLm8LF=&vdP8QS^-^lS{)5#J3gZP$RQ_!b*O- zY44eu-oBnjG(BTAti4E_nR+3%-BO|%-Jx4FEhBsJ0La(PU&dZjW74UN34Ul0So?Hg zBWA!I{uEWDo;w+>YF=HqmD-2FgkmI7I2KCyi00K{MnYn4m6+IqnCM}i{BHzDr*v34 z>%vl3UaFRuH+R%bAU%mAQ9CUMz*JmuOk`V10wA0)VmCa{OYxv=^NL()Z)12-bjdbxv>|}&Ya@PcFP~(kr&En4MBjr z5%T>qvi=I0Fc?fNi#Bnv>DjP6QxYFZ+7OY7m}y1p_J&jo+y6{K(js398)GWnAGdSL z{gqw(zvMwP&*gl$3I2adSxz`?$K&2(pUUlJI+}{Nl_!Y4KrvmADB$X`u`C}Mhu2As zLKBSOcAT#3ks+^+60dPB<-@to8(2fMLSc3t+C}g<`ewu?v(;O-VJCUj1WuNkmiHHk zUGuz|I3D5_O3xh`X||$_b)-!41!szAWJ%#p822g=SCSg)F>tLuv2)yOPMP05_kZWO z2kWkMzXQ-*{o7<6zHq60o6vVi%zVlG(^;c@Tt&;@-5+!y)0e1Ws zgWJYbd-~Q?6_|f=a~sy*rlmyf)jCtRzCNBly=8AG6DgEodX*T9!HyQc6Il36mP9Y% zD+-rvj)tR$2i!kFU9#}k!QifH;h+k5SV(S!kRhtrx?*)Pb(Vm>l*qx7_*OeXG9i22 zv?iero|)Is#*kH?^Lium*sNj_2PNO@B#I~dO-n1NRnII!`hJPMyK>2f#fyRLUPJ$B z=JJJk3zAyn2%c}wWYkqAjK!K!LAKCK&&9XLUCkXDr3V8qfW@YtOjC-wFT6lDdU;L^ zT1}fL6hjn@b*X92WWnHu3y`eiITs|2Lhx>CE7$9>JlY+{W5UVNgH>}2W2rqZ=j>hT zx{GeC??eXQVDd+QxToU90b5qk64RiG)|gy(#}l%?m^dK^`-W&j^e37K2vKR%ux|LqKN z-sm)<^dfHe$$()(KEc~=5m5=5FX&bcsg0u?`8(Ji(W)Z79+&DgqB(ObpPXlzxLqTy|4H;O*xI+fxbo%l6+nRc-_Sqk zgy7~fr0l)%C${jQNV&|<8y~006@O|~IhvqV>Gbr+IBJ!DzmTVmY8^*pq~fL4V5%%H zf`i&bFw3(wR+nyBSe>SSj`L3|nh&Z{b14x0ew1s&4OE6+n#8-2GPM&K&kktBvB54W zFOTRO=GX;M^N7*=jG>K6KmDb~k&{awdxhZ|%0R@kkN%7uj$Eq7i<+;mD=#OAsfe!y z_I@wjzb6=WXt)0k%K|0r2#q_~9}7QkGWMQaid`ad7na z*~Ndcvg5A3-iZ!CpVD90|1Z^Er%V2hY^P7`N%3QE&Q8{ejgc<@z`;A1{0|(ngZ&72 z1GditHb%Sr0h9lLgD5@}womJhzoU=e4jyG=ZBLJYoxOHsCu81AUPN>DYRR4wz#rfN z2I#Z#e^tK!Pw!j+PJ96X7XMvGz6ZdgC;;{if7hWN03dJ;07jROgP+6i$#AgWP-kZV zRvrStX9a+8Hvrr=yNQ11em{$U^Q}^Bjs*KVz1hPBcz{bl3Sfa3a0JS1qzaA!6`;Wy z0R{jDh3-6T;b1>-E;yWn1J2FKxrd9Fo0pe|n}>&wUr3OTAHmPVBPc3}5JrlKi0}%C z?Gr`r6GDn0cY;7*Yz_x}FC4xX$;ZQo{Evn83?R5b9QY9ikpWNy1crdH8iC}_-G#y+ zJ6HRkvIouyLMgS-TvS-g9%n&F7CL}DaV#a|~ z!~5n`{`yHzl{E}_VeAYMFa*#A+eKw=o^m*8dBgrh3Lt4lmo})tXqYV3igQN~&U|fk z7e3hHY4nQ?vcyRp-Q=3+n1>JI{9wn>I7W|59_i0aEKy!BZl4m=N)GAU<>GwC@}5sn z=2jl^4QnflUAUt#l2$qPO)*5|D)G|tO-Y>ghbEH!zKH(3&VmzOJrh2)lo0dl%!6V) z4>2hxLnYtl{Qr6`prQ&Iqfk+D2Bv!b2(9-^D;)Q!(2^%gz1 zb)QtPa5oP(i~5*HUP($K8Yp}wFE4ptyCi#jHwbFpd7H^Jxb2>grRMiGx}@T$a(SzF2*2>UNs19Mk<{8F|uLu z(LFi(d;#D1xti2=|LebOr7R|L-TAR4@te1DOY6#;mmzh&XKEg91U|EIVfZR8F@=jf zSRi5AJX|$d=fGWNHP5eQB*m;gsYkhEp~*M93kn@|lG5*ceN4HQ+Sf`uouz*0Zt2L- znu_>#Q|xG5@4(V&c!vAzanxh3ek6lfA0W~EbA3|mdtasXWTNeCV|0ebK%Z8K$^g8` zESrfOnRjTSnGIQ4;e!U+@#_8JTjyc^Evu#a8<|tV1HH^dqeR)(mh7}(s!`HYA;KjY zYok2>Dl}eP8n}ix4tQ4@4Ep(>50k7SHf=B~Z(qe8mbm40-cw0+-O__zlay=GpdW-C z#P3NT)UJ=~gA(rcmdEE|WA~k>rsYd3y2Tpu7YP@e6%#b?DB(j^D&(05xRTO5Bc+F+ zz4^m_nit5A5)*}lw{9n-u|UPwn+{#w>65kOh`_D|{?@ZaY!Dt~BI{(;a4&q3*)HB$hX(9y3SEyd+Jg4JKGorBw%_)~TU(Rb z5jQM8IpizYOM8(v6E#?g!DZIW{QSk^mQSQ(eOr`LvZ2H0GaKa^`RsC#?aLv~+>Lhr zWSFa6sN_PuSb1QZisLA1_mW6SD zeT}?S zrSi?|#mn~Rj%?3#wJJbYgYRkdXq~d0UaVc^*N7*(s>@4-zhsDNy-4uga8!Q4Sn##h z?+)$N2Qt4*5=T(?S>SB1o3oP}&LSBukAs+HxPQ^Se-#?OJX)d^Gh#ujUwtRc0&&ye zO$Du9;GLju+?Z9ra}(A0T{wu?44m)$37`A`S%BUl?RduYYY+6$ zVN-LJ&syJvrVu~rM4k~Y(h@mxXEXiKas|PcnAR*6fL<+>*s#n~O)j+;vARm;#NW~S zlcC~zmH|`xb!yrHYyCvJq<+Mb)V55+f-2mIX)+LJ|0yKW|U&xs5G*^t;N`wIJnX%0SAh9Mo!8dyM!bq zUQr=AuK&4%Q=ClPpR^j@IasPKN(F6ZU*Qw_xqze9O^w?XEX= zC&K936^V|;CoXqAaW=Y5wjW6PDp7deh(ej*JwEV!BG=Nj*dzA6NXb$G4t%Hxi-xUK z=n+N-^e1*dTix?2=Bg4i z*=au>{*c>kKj%=_Xu^<+xA`h8zJG!gQ8xFh1Kiq`KeMLy@z>mQ8Nv$G2d_Jb^cH2q zkmhJ|K@ygBF6a;;?ziY$D^51v`k z85n(eM6F_3+?gNa-<)ZXF`sbROX2mq?$}O5t(LP|h+>m=!LQ{;G=7)kEsuqZY( zZoIDh2Mj|lEdUD(uI8&*K_O}@JNN>`Bb|zh^kx@rj)_qji>dD3mh}rkYv#;3l9H#N zwq>y`Md`6S)kH~Ik<@17fR-PfNvMZkd7ey*e(-P4aAH!cZ)Mx~ekozCN6N5vkp&J% zAqsRAjLXfB(Br<88KdQ7zn~%r$4hfGUDVY#JAS~J+wRug-AY;F*jpD;v=-`Q=5aah z&(+i8ih24S7Y3T4-ty0WK$w*OTjyIX-C~B0{3}fL2+~G;>Sq4NsjG9_#k-+x4kpj! z4KbsYZZWf-CAbf5t8v@l$b`$bq>FCUIcsY}yvaP5OpzOIGIK0{^zB#Gi!Q98zXReN BB_#j= literal 0 HcmV?d00001 diff --git a/tests/unit/flow/test_flow_visualization.py b/tests/unit/flow/test_flow_visualization.py index 9db858d6af3d2..43752df361991 100644 --- a/tests/unit/flow/test_flow_visualization.py +++ b/tests/unit/flow/test_flow_visualization.py @@ -38,4 +38,25 @@ def test_visualization_plot_in_middle(tmpdir): with Image.open(os.path.join(cur_dir, 'flow_original_middle.jpg')) as flow_original: with Image.open(tmpfile) as flow_created: assert flow_original.size == flow_created.size - np.testing.assert_array_almost_equal(flow_original, flow_created) \ No newline at end of file + np.testing.assert_array_almost_equal(flow_original, flow_created) + + +def test_visualization_plot_twice(tmpdir): + cur_dir = os.path.dirname(os.path.abspath(__file__)) + tmpfile_middle = os.path.join(tmpdir, 'flow_test_middle.jpg') + tmpfile_end = os.path.join(tmpdir, 'flow_test_end.jpg') + + flow = (Flow().add(name='pod_a') + .plot(output=tmpfile_middle) + .add(name='pod_b', needs='gateway') + .join(needs=['pod_a', 'pod_b']).plot(output=tmpfile_end)) + + with Image.open(os.path.join(cur_dir, 'flow_original_middle.jpg')) as flow_original_middle: + with Image.open(tmpfile_middle) as flow_created_middle: + assert flow_original_middle.size == flow_created_middle.size + np.testing.assert_array_almost_equal(flow_original_middle, flow_created_middle) + + with Image.open(os.path.join(cur_dir, 'flow_original_end.jpg')) as flow_original_end: + with Image.open(tmpfile_end) as flow_created_end: + assert flow_original_end.size == flow_created_end.size + np.testing.assert_array_almost_equal(flow_original_end, flow_created_end) \ No newline at end of file