g"<ddlZddlZddlmZddlmZddlmZ dZ edgdZ edgdZ dZ ed gd ejd krdnd ed dZedgd ejd krdnd eddZedgd ddZedgdd ejd krdnddZedgdd dZd ZdS)!N)image_comparisonc d|ddd|zfddt|dddS) N?rrbk) arrowstylefcec)xyxytextsize arrowprops)annotatedict)axtrs Y/granary/edev/quantum/lib/python3.11/site-packages/matplotlib/tests/test_arrow_patches.py draw_arrowrsPKKS1W~jr1#&300011111fancyarrow_test_imagec gd}ddtjg}tjt |t |ddt d\}}t|D]L\}}t|D]7\}}|||f}t|||| dd 8MdS) N)皙?333333?皙?皙?rfancysimpleF)g@)aspect)squeezefigsize subplot_kw) labelleft labelbottom) mpatches ArrowStyleFancypltsubplotslenr enumerater tick_params) rrfigaxsi_rr1i_tt1rs rtest_fancyarrowr8s A (H/55778A|CFFCFFE$,QIIIHCQ<<??R || ? ?GCS#XB r2r " " " NNUN > > > > ???rzboxarrow_test_image.pngcjtj}t|}d}||zdz}t jd|dz f}d}t t|D]B\}}|d||z |zdz |z |d||j t|dd  CdS) Ng333333?rgUUUUUU@g?r&g5@centerwr )boxstyler r )har transformbbox) r*BoxStyle get_stylesr/r-figurer0sortedtext transFigurer)stylesnspacing figheightr2fontsizei stylenames r test_boxarrowrNs  ) ) + +F F AGWr!I *gy37 8 8 8CH!&..11@@ 9 A(3. 99?I##>>>  @ @ @ @@@rc tjddd}|}|ddg|ddg|t jddd d d |S) z Convenience function that prepares and returns a FancyArrowPatch. It aims at being used to test that the size of the arrow head does not depend on the DPI value of the exported picture. NB: this function *is not* a test in itself! fancyarrow_dpi_cor_test)2)r&dpirr#)rr)皙?g333333?rR->d)posAposBlwr mutation_scale)r-rC add_subplotset_xlimset_ylim add_patchr*FancyArrowPatch)fig2rs r!__prepare_fancyarrow_dpi_cor_testrb3s :/R H H HD    BKKAKKALL)z -.49<>>>??? Krzfancyarrow_dpi_cor_100dpi.pngTx86_64g{Gz?rW)rT) remove_texttol savefig_kwargc"tdS)a Check the export of a FancyArrowPatch @ 100 DPI. FancyArrowPatch is instantiated through a dedicated function because another similar test checks a similar export but with a different DPI value. Remark: test only a rasterized format. Nrbrrtest_fancyarrow_dpi_cor_100dpirjEs&'''''rzfancyarrow_dpi_cor_200dpi.pngc"tdS)z As test_fancyarrow_dpi_cor_100dpi, but exports @ 200 DPI. The relative size of the arrow head should be the same. Nrhrirrtest_fancyarrow_dpi_cor_200dpirmTs&'''''rzfancyarrow_dash.pngdefault)rdstylec tj\}}tjdddddddd }tjdddd ddd d }||||dS) N)rrrz-|>zangle3,angleA=0,angleB=90g$@dashedr )r connectionstyler[ linewidth linestylecolorangle3dotted)r-r.r*r`r_)r2ree2s rtest_fancyarrow_dashr{`slnnGC ,11L04+,+3'*  , , ,A  !&*-22:15,-,4(+  - - -BLLOOOLLrzarrow_styles.pngmpl20g{Gzt?)rordrecHtj}t|}t jd\}}|dd|d||ddddtt|D]H\}}tj d|dzd zz|fd |dzd zz|f|d }| |Itgd D]\}}|}|ddkr|dz }|ddkr|dz }tddgD]e\}} | dt| } tj dd|z|zfdd|z|zf| d }| |fdS)N)r"r"r:rr#)leftrightbottomtoprrqg?g?)r r[)z]-[z]-z-[z|-|-z ,angleA=ANGLEz ,angleB=ANGLEi<ANGLEg?g?)r*r+rBr/r-r.r]r^subplots_adjustr0rDr`r_replacestr) rGrHr2rrLrMpatchrojangler s rtest_arrow_stylesrus + + - -F F Al6***GCKK1KKAQaq999!&..11 9(#Q *@@@E LL         rzconnection_styles.png)rordctj}t|}t jd\}}|dd|d|tt|D]:\}}tj d|fd|dzfd |d }| |;dS) N) r:rr#r~rrUrrVr)r rsr[) r*ConnectionStylerBr/r-r.r]r^r0rDr`r_)rGrHr2rrLrMrs rtest_connection_stylesrs  % 0 0 2 2F F Al7+++GCKK1KKA!&..11 9(#qCS>489B8:<<< U rctjdd}tjdd|}t jt 5tj |dddn #1swxYwYtjdd}tjdd|}tj |dS)Nrk)angleAangleB)rrr)rsgh@) r*rAngle3r`pytestraises ValueErrorr-gcar_) conn_style_1p1 conn_style_2p2s rtest_invalid_intersectionrs+22"S2IIL  !(H2> @ @ @B z " "    B               +22"U2KKL  !(H2> @ @ @BGIIs'BB  B )rplatformmatplotlib.pyplotpyplotr-matplotlib.testing.decoratorsrmatplotlib.patchespatchesr*rr8rNrbmachinerrjrmr{rrrrirrrs` ::::::%%%%%%111 *+,, ? ?-, ?,-..@@/.@($23+8+--99aat $ /// ( (// (23+8+--99aat $ ///((//(()t9MMMNM(%&g4+8+--99aauFFF  FF >*+7MMM  NM      r