U md<@shddlZddlmZmZddlmZddlmZddZddZ d d Z ej d d d idfddddfdd idfdd idfdd gd ddgddfdd gdddgddffddZ ej d d d id!fd"d#gdd$d%fd&d id'fd&did(fd)dd*d+fd,dd*d-fd.dd*d/fd0dd1d"gdgd2d3fd0dd1d"gdgd4d5ff d6d7Zd8d9Zed:gd;dN)assert_allcloseassert_array_equal)Sankey)check_figures_equalcCst}|dS)N)radd)sankeyrU/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/matplotlib/tests/test_sankey.py test_sankeysr cCs4tdgdgdgd}|jdjddks0tdS)N?First)flowslabels orientationsrz First 0.25rZdiagramstextsget_textAssertionError)srrr test_labelsrcCs>dd}tdgdgdg|d}|jdjddks:tdS) 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 negativerrz/The shapes of 'flows' \(2,\) and 'orientations'abc)rrz)The shapes of 'flows' \(2,\) and 'labels'c Cs(tjt|dtf|W5QRXdS)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'c Cs@t}tjt|d"|jddgd|jf|W5QRXdS)Nr"r*r.)r)rr$r%r&r)r'r(rrrr test_sankey_add_errors.sr/cCs8tddddgdgdgdd}|}t|d jddddg|d jd d d d gksVttd d |d jDsrttdd |d jDst|d j dkstt |d j ddddgtddd ddgdgdgdd}|}t|d jddd ddg|d jd d dd d gkstt |d j dddddgdS)Nr п?Foor Bar)rrrunitrrr-cSs g|]}|dddkqS)rr-r3r.0textrrr Hsz test_sankey2..cSs g|]}|dddkqS)Nr4r6r7rrr r:Is)gg=ä)g?g6 t))ggՁA ڿ)g?gqXu)rr) rfinishrrZanglesrallrr9rrZtips)rZsfrrr test_sankey2Bs@  r?Zpng) extensionsc Csz|}t|ddddddgddddddgd}||}t|d }|jddddddgddddddgd |dS) Nr r0r1r2rr r)axrr)rAr)Zgcarr=r)Zfig_testZfig_refZax_testZs_testZax_refZs_refrrr test_sankey3^s rB)r$Z numpy.testingrrZmatplotlib.sankeyrZmatplotlib.testing.decoratorsrr rrmarkZ parametrizer)r/r?rBrrrr sT