a d<@sdddlZddlmZmZddlmZddlmZddZddZ d d Z ej d d d idfddddfdd idfdd idfdd ggdddfdd ggdddffddZ ej d dd id fd!d"gdd#d$fd%d id&fd%did'fd(dd)d*fd+dd)d,fd-dd)d.fd/dd0d!gdgd1d2fd/dd0d!gdgd3d4ff d5d6Zd7d8Zed9gd:d;d<ZdS)=N)assert_allcloseassert_array_equal)Sankey)check_figures_equalcCst}|dS)N)radd)sankeyre/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/matplotlib/tests/test_sankey.py test_sankeysr cCs4tdgdgdgd}|jdjddks0JdS)N?First)flowslabels orientationsrz First 0.25rZdiagramstextsget_text)srrr test_labelsrcCs>dd}tdgdgdg|d}|jdjddks:JdS) NcSs|dS)Nz.3fr)valuerrr show_three_decimal_placessz=test_format_using_callable..show_three_decimal_placesr r r )rrrformatrz First 0.250r)rrrrr test_format_using_callables rz kwargs, msggapr z'gap' is negative)rZradiusz'radius' is greater than 'gap'Z head_anglez'head_angle' is negativeZ tolerancez'tolerance' is negative)r rrrrz/The shapes of 'flows' \(2,\) and 'orientations')abc)rrz)The shapes of 'flows' \(2,\) and 'labels'cCs@tjt|dtfi|Wdn1s20YdS)Nmatch)pytestraises ValueErrorr)kwargsmsgrrr test_sankey_errorss r(Z trunklengthz'trunklength' is negative皙?g333333?)rpriorzThe scaled sum of the connectedr*z*The index of the prior diagram is negativez#The index of the prior diagram is 1)r r)connectr*zAt least one of the connection)rrz"The connection index to the source)rz The connection index to this dia)rr皙ɿ)r+r*rrzThe value of orientations)r+r*rZ pathlengthszThe lengths of 'flows'cCsXt}tjt|d0|jddgd|jfi|Wdn1sJ0YdS)Nr!r)r-)r)rr#r$r%r)r&r'rrrr test_sankey_add_errors.sr.cCstgddgdgdd}|}t|djgd|djgdksJJtdd |djDsfJtd d |djDsJ|djd ksJt |dj gd tgd dgdgdd}|}t|djgd |djgdksJt |dj gddS)N)r п?Foor Bar)rrrunitr)rr,rr,cSs g|]}|dddkqS)rr,r2r.0textrrr Hz test_sankey2..cSs g|]}|dddkqS)Nr3r5r6rrr r9Ir:)gg=ä࿩g?g6 t)迩ggՁA ڿg?gqXu)r r/rr0r1)rr,Nrr,)r=r>)rrr?r@) rfinishrrZanglesallrr8rrZtips)rZsfrrr test_sankey2Bs*    rCZpng) extensionscCsZ|}t|gdgdd}||}t|d}|jgdgdd|dS)N)r r/r/r r0r1)rr rr rr)axrr)rEr)ZgcarrAr)Zfig_testZfig_refZax_testZs_testZax_refZs_refrrr test_sankey3^s   rF)r#Z numpy.testingrrZmatplotlib.sankeyrZmatplotlib.testing.decoratorsrr rrmarkZ parametrizer(r.rCrFrrrr sT