o GfJ @s<ddlZddlZddlmZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZddlmZd d Zd d Zd dZddZejde gdddddfe gdddddfe gdddddfe gdddddfe gdddddfe gdddddfgddZddZe ddgddgddgddgge je je je jge ddgddgddgge je je jge ddgddgge je jge dd gge jggZ gd!Z!ejd"e"e e!d#d$Z#ejd%e j$e j%gd&d'Z&d(d)Z'd*d+Z(e d,gdd-d.d/d0Z)e d1gdd2d3d4Z*e d5gd-d6d7d8Z+d9d:Z,d;d<Z-e d=gdd2d>d?Z.e d@gdd2dAdBZ/e dCgdd2dDdEZ0e dFgdGdgdHdIdJdKZ1dLdMZ2dNdOZ3dPdQZ4dRdSZ5ejdTe6dUdVdWDdXdYZ7ejdZe8d[d\d]d^d_Z9d`daZ:dbdcZ;dddeZ.modify_vertices)r unit_circlepytestraisesAttributeError)rrrrtest_readonly_paths   "rcCs"tddd}tjttd|jd t |Wdn1s&wYtdddd}tjttd|jd t |Wdn1sRwYtddd}td}td|jd |j}tjt|dt ||WddS1swYdS) N z has shape )matchr zYour vertices have shape z but your codes have shape ) r arangeZreshaperr ValueErrorreescaperr)Z bad_verts1Z bad_verts2Z good_vertsZ bad_codesmsgrrrtest_path_exceptions#s(    "r*cCsDtgd}ddg}||}|jdksJtj|ddgdS)N)rrrr-r-r-r?r1)?r1boolTF)rZ_create_closedcontains_pointsdtyper testing assert_equal)rpointsretrrrtest_point_in_path6s  r:z#other_path, inside, inverted_inside)?r<)r<?)r=r=)r=r<r;Tr F)пr?r??rArArAr?r>)r>r@r0rBrCr>)r;)r<?)rDrD)rDr<r;r+r,r.r/r+)r r )r r"r"r")r"r rFcCs8tgddd}|||usJ|||usJdS)NrETr )rZ contains_path)Z other_pathinsideZinverted_insiderrrrtest_contains_path?srIcCs4t}gd}|j|dd}tj|gddS)N))rJ)rDrJ)?rKg)Zradius)TFF)rrr4r r6r7)rr8resultrrr$test_contains_points_negative_radiusRsrMr-r ))rJrJr=?)rJrJrNr1)rJrNrNrN)rNrrNrz path, extentscCst|j|ks JdS)N)r allrr)rrrrrtest_exact_extentsks rP ignored_codecCsBtddgddgddggtjtj|g}t|jdksJdS)Nrr-r )rJrJrNrN)rMOVETOr rOrr)rQrrrrtest_extents_with_ignored_codesys rScCsltddgddgddgddgddgg}t|}ttjdgg}||}t|dks.J|dr4JdSNrr-r1)r arrayrnanr4len)Zboxptestcontainsrrrtest_point_in_path_nans ( r[cCst\}}|jddd|dd}||jd|js"J||jd|jr1J||jd|jr@JdS) Nlogr,)Zxscaleylimr- )r1r0)2r1)pltsubplotssetZaxvspanZget_pathZcontains_point transData transform)figaxZpolygonrrrtest_nonlinear_containments   rhzarrow_contains_point.pngZmpl20) remove_textstylecCs&t\}}|d|dtjddddd}||tjddd dd}||tjd d d d dd}|||||g}tt dddt ddd\}}t t | | D]3\}\} } |j | | g\} } td|j| | } t |D]\}}|| \}}|r|j| | dddqyq]dS)Nr )r1r<)r2r=z->() arrowstylemutation_scale)r1r-)r2rDz]-[)r1r2)r2rAZfancyF)rlfillrmrr g?Zbutton_press_eventr_r)sc)rarbset_xlimset_ylimrZFancyArrowPatch add_patchr Zmeshgridr% enumeratezipZravelrdrerZcanvasrZZscatter)rfrgZarrowZarrow1Zarrow2Z patches_listXYkxyZxdispZydispZeventmZpatchrHresrrrtest_arrow_contains_pointsB        "r~zpath_clipping.svg)ric Cstjdd}tgdgdgdgdgdgdgd gD]8\}}|d d |d }gd }||d|d|d ||d |d |d|t|ddddqdS)N)g@g@)Zfigsize)rr^r)rr)rrrrd)rrrrr)rrrr)rrrrr)rrrr)rrrrr!r r-)riirr"ZnoneZredT)Z facecolorZ edgecolorr )raZfigureruZ add_subplotrrrsrtr)rfiZxyrgZbboxrrrtest_path_clippings$  rzsemi_log_with_zero.png)rjcCs^tdd}dd|ddd}t\}}|j||ddd d |d d|d dS) Nir^rNr r--ory)ZlwZmarkeredgecolorgHz>T)r r%rarbZsemilogyrsZgridrzr{rfrgrrrtest_log_transform_with_zeros   rcCszt}|jjdks Jt||}|jjdksJ|jjdks"Jttdg|}|jjdks3J|jjdks;JdS)Nr )rr+)r-r )r-)rmake_compound_pathrrr)emptyZr2Zr3rrrtest_make_compound_path_emptys rcCsLddg}dt||gtjtjgg}tj|}t|jtjkdks$JdS)Nrr")rrRSTOPrr sumr)ZzeropathsZ compound_pathrrrtest_make_compound_path_stopss rzxkcd.pngcCsptjdtddtjd}t|}tt\}}| ||WddS1s1wYdS)Nrr r) r randomseedlinspaceZpiZsinraxkcdrbplotrrrr test_xkcds    "rzxkcd_marker.pngcCstjdtddd}|}d|}dtd}t*t\}}|j||ddd|j||ddd|j||d ddWddS1sKwYdS) Nrr_@+r^)mso^) r rrronesrarrbr)rzy1Zy2Zy3rfrgrrrtest_xkcd_markers   "rzmarker_paths.pdfcCsDd}tt|t|dt|td|tdddS)Nr!)raZerrorbarr r%rZxlimr])Nrrrtest_marker_paths_pdfs   rZnan_pathdefault)ZpdfZsvgepsZpng)rjri extensionscCsVdtjdtjdddg}tjdtjddtjd g}t\}}||d ||d dS) Nrr r!r_r$r r^r r)r rVrarbr)Zy0rrfrgrrrtest_nan_isolated_pointss   rcCs*tddgddgddgddgdd gd d gdd gd dgddgddgddgddgddgddgddgddgdd gd!d"gd#d"gd$d%gd&d'gd(d)gg}d*\}}}}tt|ddd+f|ddd,ffjd-d.}t||g||gg}||d,}t |d/|d0ksJt |d0|d,ksJdS)1NgmR?gDpTU?g?g_9?g?g?gUUu@gF^9?g?g$z|?g["8j@gUU@ga1?g["8@g}LRJ?gDpTU?gmR?g@g=:r9?g["8@gAL0@gUU@gy?gI?g9@g["8 @g_9@g@gB@gUU@gq1Pc1@g@?gh s@gUU@gA!@gmR@g@gbc@g@?g9qa@gI?)rNr2g@rr-rTr r) r rUrZvstackTrrZ clip_to_bbox to_polygonsrO)ZhandZr0Zc0Zr1Zc1ZpolyZ clip_rectZ poly_clippedrrr(test_path_no_doubled_point_in_to_polygon(s> 0rcCsddgddgg}t|}t|jdddgt|jdddd|gt|gt|jdd|gddgddgddgg}ddgddgddgddgg}t|}t|jddd|gt|jdddd|gt||gt|jdd|gdS) Nr^rrk)widthheightF)rr closed_only)r)rrr)datarXZ closed_datarrrtest_path_to_polygonsKs"rcCsddgddgg}tjtjg}t|}t||}|}|}||us%J|j|jus-J||us3J|j|jus;J|j|jusCJdSNrr-)rrRLINETOZdeepcopyrrZvertsrZpath1Zpath2Z path1_copyZ path2_copyrrrtest_path_deepcopy`    rcCsddgddgg}tjtjg}t|}t||}|}|}||us%J|j|jus-J||us3J|j|jus;J|j|jusCJdSr)rrRrcopyrrrrrrtest_path_shallowcopyorrphicCsg|] }tgd|qS)) rrr-<KZix)r rU).0Zdeltarrr ~sr)rrr-cCsxgd}tt|}tddg}||}||r#||s%Jtddg}||}||r:||sg:0yE>g|=g-q=)rr)r rr+r,)rr")r-r"rr_)r_r_r.rG)r"r)rr$)rrrr-r_)r,r r)r-r)rr)rKr-)rJ)rNr)rN@)rJr)rZAffine2Drotater Zdeg2radrZtransform_pathintersects_path)rZ eps_arrayreabrrrrtest_path_intersect_path~sZ    "   roffseti0iircCsX|}d|}t||}tj|jdd}tj|jdd}tj|dtj|ddS)Nihr)Zaxisrr-)rZarcr minrmaxr6Zassert_allclose)rZlowZhighrZminsZmaxsrrr test_full_arcs rc Csttddgddgddgddgddggtjgd tjd }ttd d gd dgd dgd d gd d ggtjgdtjd }||rEJ||rLJdS)NgƉ@g퟇@gЅ@g@g`m$@g;E@g6[@gч61@)r-r r r O)r5g-Hߊ@gGN@g`؊@g{Kʠ@g @)r-r r r r )rr rUZ code_typerZ this_pathZ outline_pathrrr!test_disjoint_zero_length_segments0   rcCs`ttddgddgg}ttddgddgddgddgg}||s'J||s.JdSrT)rr rUrrrrr"test_intersect_zero_length_segments rcCsttjtjgtjtjggtjtjgttjtjgtjtjggttjtjgtjtjgtjtjgtjtjggtjtjtjtjgg}|D]}|jdd}t|dksSJ|jdtj ks]JqCdS)NT)Z remove_nansr-r) rr rVrR CLOSEPOLYCURVE3cleanedrWrr)rrXrrrrtest_cleanup_closepoly s    r)=r'Znumpyr Z numpy.testingrrZ matplotlibrZmatplotlib.pathrZmatplotlib.patchesrZmatplotlib.testing.decoratorsrZmatplotlib.pyplotZpyplotrarZmatplotlib.backend_basesrrrr*r:ZmarkZ parametrizerIrMrRZCURVE4rrZ _test_pathsZ_test_path_extentsrvrPrrrSr[rhr~rrrrrrrrrrrrZ concatenaterrangerrrrrrrrs            &        %           # Q