U d{j@s`ddlZddlZddlmZddlmZddZddZdd Z ej d d d id fddddfdd idfdd idfdd gd ddgddfdd gdddgddffddZ 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 d5d6Z d7d8Zed9gd:d;d<ZdS)=N)Sankey)check_figures_equalcCst}|dS)N)radd)sankeyr[/var/www/html/bwave/python/venv/lib/python3.8/site-packages/matplotlib/tests/test_sankey.py test_sankeysrcCs4tdgdgdgd}|jdjddks0tdS)N?First)flowslabels orientationsrz First 0.25rZdiagramstextsget_textAssertionError)srrr test_labelsrcCs>dd}tdgdgdg|d}|jdjddks:tdS) NcSs|dS)Nz.3fr)valuerrrshow_three_decimal_placessz=test_format_using_callable..show_three_decimal_placesr r r )r r rformatrz First 0.250r)rrrrrtest_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 negativer rz/The shapes of 'flows' \(2,\) and 'orientations'abc)r r z)The shapes of 'flows' \(2,\) and 'labels'c Cs(tjt|dtf|W5QRXdS)Nmatch)pytestraises ValueErrorr)kwargsmsgrrrtest_sankey_errorss r'Z trunklengthz'trunklength' is negative皙?g333333?)r priorzThe 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)r rzThe value of orientations)r*r)r Z pathlengthszThe lengths of 'flows'c Cs@t}tjt|d"|jddgd|jf|W5QRXdS)Nr r(r,)r )rr"r#r$r)r%r&rrrrtest_sankey_add_errors.sr-cCsTtddddgdgdgdd}|}tttd |d jsFt|d jd d d d gks`ttd d|d j Ds|ttdd|d j Dst|d j dkstt td|d j sttddd ddgdgdgdd}|}tttd|d jst|d jd d dd d gks2tt td|d j sPtdS)Nr п?Foor Bar)r r runit)r r.r/r0rrr+cSs g|]}|dddkqS)rr+r1r.0textrrr Hsz test_sankey2..cSs g|]}|dddkqS)Nr2r4r5rrrr8Is)gg=ä࿩g?g6 t)迩ggՁA ڿg?gqXu)r r.rr/r0)r;r<)rrr=r>)rfinishnpallequalarrayr rZanglesrr7rZallcloseZtips)rZsfrrr test_sankey2Bs." $ rDZpng) extensionsc Csz|}t|ddddddgddddddgd}||}t|d }|jddddddgddddddgd |dS) Nr r.r/r0rr r)axr r)rFr)Zgcarr?r)Zfig_testZfig_refZax_testZs_testZax_refZs_refrrr test_sankey3^s rG)r"numpyr@Zmatplotlib.sankeyrZmatplotlib.testing.decoratorsrrrrmarkZ parametrizer'r-rDrGrrrrsT