g<ddlZddlmZmZddlmZddlmZdZdZ dZ ej dd d id fd d ddfdd idfdd idfd d ggdddfd d ggdddffdZ ej ddd idfddgddd fd!d id"fd!d id#fd$dd%d&fd'dd%d(fd)dd%d*fd+dd,dgd gd-d.fd+dd,dgd gd/d0ff d1Zd2Zed3g4d5ZdS)6N)assert_allcloseassert_array_equal)Sankey)check_figures_equalcJt}|dS)N)radd)sankeys R/granary/edev/quantum/lib/python3.11/site-packages/matplotlib/tests/test_sankey.py test_sankeyr s XXF JJLLLLLctdgdgdg}|jdjddksJdS)N?First)flowslabels orientationsrz First 0.25rdiagramstextsget_text)ss r test_labelrsMdVWIRDAAAA :a= q ! * * , , = = = = = =r cd}tdgdgdg|}|jdjddksJdS)Nc |dS)Nz.3f)values r show_three_decimal_placesz=test_format_using_callable..show_three_decimal_placess ~~r rrr)rrrformatrz First 0.250r)rrs r test_format_using_callabler si dVWIRD/ 1 1 1A :a= q ! * * , , > > > > > >r z kwargs, msggaprz'gap' is negative)r!radiusz'radius' is greater than 'gap' head_anglez'head_angle' is negative tolerancez'tolerance' is negative)rrr"rrz/The shapes of 'flows' \(2,\) and 'orientations')abc)rrz)The shapes of 'flows' \(2,\) and 'labels'ctjt|5tdi|ddddS#1swxYwYdS)Nmatchr)pytestraises ValueErrorr)kwargsmsgs r test_sankey_errorsr3s z - - -s 6:: trunklengthz'trunklength' is negative皙?g333333?)rpriorzThe scaled sum of the connectedr6z*The index of the prior diagram is negativez#The index of the prior diagram is 1)rr")connectr6zAt least one of the connection)r#r"z"The connection index to the source)r"z The connection index to this dia)r"r"皙ɿ)r7r6rrzThe value of orientations)r7r6r pathlengthszThe lengths of 'flows'ct}tjt|5|ddg|jdi|ddddS#1swxYwYdS)Nr,r5r9)rr)rr.r/r0r)r1r2r s r test_sankey_add_errorsr<.sXXF z - - - #t %%% Vs&AA!$A!ctgddgdgd}|}t|djgd|djgdksJt d|djDsJt d |djDsJ|djd ksJt|dj gd tgd dgdgd}|}t|djgd |djgd ksJt|dj gddS)N)rп?FoorBar)rrrunitr)r"r8r"r8cNg|]"}|dddk#S)rr8rAr.0texts r z test_sankey2..Hs.FFF$ !$-FFFr cNg|]"}|dddk#S)NrBrErFs r rIz test_sankey2..Is.FFF$ $-FFFr )gg=ä࿩g?g6 t)迩ggՁA ڿg?gqXu)rr>rr?r@)r"r8Nr"r8)rMrN)rrrOrP) rfinishrranglesallrrHrrtips)rsfs r test_sankey2rVBs---ugDu . . .A Br!u{$<$<$<=== a5<<<< ' ' ' ' FF"Q%+FFF G GGGG FF"Q%+FFF G GGGG a5:   B & & & &BqEJ)))*** 000%Du . . .A Br!u{$?$?$?@@@ a5<--- - - - -BqEJ)))*****r png) extensionsc(|}t|gdgd}||}t|}|gdgd|dS)N)rr>r>rr?r@)r"rr"rrr)axrr)rZr')gcarrQr)fig_testfig_refax_tests_testax_refs_refs r test_sankey3rb^sllnnG w&K&K&K!5!5!5777F MMOOO [[]]F f   E II999///111 LLNNNNNr )r. numpy.testingrrmatplotlib.sankeyrmatplotlib.testing.decoratorsrr rr mark parametrizer3r<rVrbrr r rhsN ========$$$$$$====== >>> ? ? ? R[%& !>?B342122w 33792w///2213 )     R56SzA&&(IJ r]@A q\891%%'GH!$$&JK!$$&HI!tSkc79!tSkS35 )    ***8(((  )(   r