o
    l’×iGt  ã                   @   sÚ  d dl Z d dlZd dlmZmZmZmZ d dlZd dl	m
Z
 d dlmZ d dlmZ d dlmZ d dlmZmZmZ d dlmZ d dlmZmZ dd„ Zd	d
„ Zedgdddddd„ ƒZdd„ Zdd„ Z dd„ Z!dd„ Z"dd„ Z#dd„ Z$G dd„ dej%ƒZ&G d d!„ d!ƒZ'G d"d#„ d#ƒZ(d$d%„ Z)d&d'„ Z*d(d)„ Z+d*d+„ Z,d,d-„ Z-d.d/„ Z.d0d1„ Z/d2d3„ Z0d4d5„ Z1d6d7„ Z2d8d9„ Z3d:d;„ Z4d<d=„ Z5ej6 7d>g d?¢¡d@dA„ ƒZ8dBdC„ Z9dDdE„ Z:dFdG„ Z;edHgdIdJdK„ ƒZ<dLdM„ Z=dNdO„ Z>dS )Pé    N)Úassert_allcloseÚassert_almost_equalÚassert_array_equalÚassert_array_almost_equal)Úscale)ÚAffine2DÚBboxÚTransformedBbox)ÚPath)Úimage_comparisonÚcheck_figures_equalc                  C   s^   G dd„ dt jƒ} | ƒ }t ¡ }tjt d¡||j d t ¡  d|_	|j
 ¡  t ¡  d S )Nc                       sD   e Zd ZdZd ZZdZ‡ fdd„Zdd„ ZeZ	dd	„ Z
e
Z‡  ZS )
z<test_non_affine_caching.<locals>.AssertingNonAffineTransformz
        This transform raises an assertion error when called when it
        shouldn't be and ``self.raise_on_transform`` is True.

        é   Fc                    s.   t ƒ j|i |¤Ž d| _t ¡  dd¡| _d S )NFé
   )ÚsuperÚ__init__Úraise_on_transformÚmtransformsr   r   Úunderlying_transform)ÚselfÚargsÚkwargs©Ú	__class__© ú^/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/matplotlib/tests/test_transforms.pyr      s   zEtest_non_affine_caching.<locals>.AssertingNonAffineTransform.__init__c                 S   ó   | j rJ dƒ‚| j |¡S ©Nz3Invalidated affine part of transform unnecessarily.)r   r   Útransform_path©r   Úpathr   r   r   Útransform_path_non_affine    ó   ÿzVtest_non_affine_caching.<locals>.AssertingNonAffineTransform.transform_path_non_affinec                 S   r   r   )r   r   Ú	transformr   r   r   r   Útransform_non_affine&   r!   zQtest_non_affine_caching.<locals>.AssertingNonAffineTransform.transform_non_affine)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ú
input_dimsÚoutput_dimsÚ	is_affiner   r    r   r#   r"   Ú__classcell__r   r   r   r   ÚAssertingNonAffineTransform   s    r,   r   ©r"   T)r   Ú	TransformÚpltÚaxesÚplotÚnpÚarangeÚ	transDataÚdrawr   Ú	transAxesÚ
invalidate)r,   Úmy_transÚaxr   r   r   Útest_non_affine_caching   s   
r:   c                  C   sn   G dd„ dƒ} t  ¡ }t jt d¡| dƒd\}| dd¡ | dd¡ t| ¡ j	 
¡ t ¡  d¡ 
¡ ƒ d S )Nc                   @   s   e Zd Zdd„ Zdd„ ZdS )z-test_external_transform_api.<locals>.ScaledByc                 S   s
   || _ d S ©N)Ú_scale_factor)r   Úscale_factorr   r   r   r   9   s   
z6test_external_transform_api.<locals>.ScaledBy.__init__c                 S   s   t  ¡  | j¡|j S r;   )r   r   r   r<   r4   )r   r0   r   r   r   Ú_as_mpl_transform<   s   ÿz?test_external_transform_api.<locals>.ScaledBy._as_mpl_transformN)r$   r%   r&   r   r>   r   r   r   r   ÚScaledBy8   s    r?   r   r-   r   éd   )r/   r0   r1   r2   r3   Úset_xlimÚset_ylimr   Úget_transformÚ_aÚ
get_matrixr   r   r   )r?   r9   Úliner   r   r   Útest_external_transform_api7   s   ÿrG   Úpre_transform_dataTÚmpl20gš™™™™™©?)Úremove_textÚstyleÚtolc               
   C   s¼  t  ¡ } t ¡  d¡}| jt d¡ dd¡|| j	 d | j
t ddd¡t d	dd
¡t d¡ dd¡|| j	 d | jt dd¡t dd¡|| j	 d t ddd¡}t ddd¡}dt |¡ t |d d …tjf ¡ }t |¡t |d d …tjf ¡ }| j|||||| j	 t ||¡d |d d d… |d d d… }}|d d d…d d d…f |d d d…d d d…f }}| j||d |||| j	 d | j|d |d |d |d || j	 d d S )Nr   é0   é   é   r-   r   é   é   g      @é	   é   é   é   r   )r"   Ú	linewidthé   )r/   r0   r   r   r   Úcontourfr2   r3   Úreshaper4   Ú
pcolormeshÚlinspaceÚscatterÚsinÚcosÚnewaxisÚ
streamplotÚhypotÚquiverÚbarbs)r9   Útimes10ÚxÚyÚuÚvr   r   r   Útest_pre_transform_plottingI   s,   "ýÿ&"
ÿ2.ri   c                  C   s‚   t  ¡ } t t ddd¡t ddd¡¡\}}| j||t || ¡t ¡  	d¡| j
 d t ddgd	d
gg¡}t|| j ¡ ƒ d S )Né   rS   çÍÌÌÌÌÌ(@ç      )@çš™™™™™¹?r-   ç      ø?ç×£p=
×ó?ç       @ç      ô?)r/   r0   r2   Úmeshgridr[   rX   Úlogr   r   r   r4   Úarrayr   ÚdataLimÚ
get_points©r9   ÚxsÚysÚexpectedr   r   r   Ú!test_contour_pre_transform_limitsn   s   $ÿ
ÿr{   c                  C   ó’   t  ¡ } t t ddd¡t ddd¡¡\}}| j||t || ¡d d…d d…f t ¡  	d¡| j
 d t dd	gd
dgg¡}t|| j ¡ ƒ d S ©Nrj   rS   rk   rl   éÿÿÿÿrm   r-   rn   ro   rp   rq   )r/   r0   r2   rr   r[   Úpcolorrs   r   r   r   r4   rt   r   ru   rv   rw   r   r   r   Ú test_pcolor_pre_transform_limitsy   ó   $$ÿ
ÿr€   c                  C   r|   r}   )r/   r0   r2   rr   r[   rZ   rs   r   r   r   r4   rt   r   ru   rv   rw   r   r   r   Ú$test_pcolormesh_pre_transform_limits…   r   r‚   c                  C   sÄ  t  ddgddgddgg¡} tj dddddd¡}| | ¡}t  ddgddgddgg¡}t||ƒ tj dddddd¡}| | ¡}t  ddgddgddgg¡}t||ƒ tj dddddd¡}| | ¡}t  ddgd	dgddgg¡}t||ƒ tj dddd
dd¡}| | ¡}t  ddgddgddgg¡}t||ƒ tj dddddd¡}| | ¡}t  ddgddgddgg¡}t||ƒ tj dddddd¡}| | ¡}t  ddgddgddgg¡}t||ƒ d S )Nr   r   rS   r~   rT   r   éþÿÿÿrW   é<   rP   éP   rU   rN   )r2   rt   r   r   Úfrom_valuesr"   r   )ÚpointsÚtÚactualrz   r   r   r   Útest_Affine2D_from_values‘   s8   
þ










rŠ   c                  C   sX   ddg} t  ¡ }t| | | ¡  | ¡¡ƒ | dd¡ ¡  t| | | ¡  | ¡¡ƒ d S )Nç      ð?)r   r   r   r"   ÚinvertedÚ	translaterE   )Úpointrˆ   r   r   r   Ú test_affine_inverted_invalidated¶   s
   r   c                  C   sh   t  g d¢¡} t t ¡ t dd¡¡}| | ¡}|j| 	¡ ddd}t
|Ž \}}t|| jd d… ƒ d S )N))çš™™™™™É?éÿÿÿ)çš™™™™™Ù?r‘   )r’   rS   )r   rS   r   Úclip)r   r   r@   r@   F)r“   Úsimplifyr~   )r
   Ú_create_closedr   ÚBlendedGenericTransformr   r   ÚLogTransformr    Úiter_segmentsÚ
get_affineÚzipr   Úcodes)r   ÚtransÚtpathÚresultÚtpointsÚtcodesr   r   r   Útest_clipping_of_logÁ   s   ÿ

þr¡   c                       s<   e Zd ZdZdZdZdZ‡ fdd„Zdd„ Zdd	„ Z	‡  Z
S )
ÚNonAffineForTestzÚ
    A class which looks like a non affine transform, but does whatever
    the given transform does (even if it is affine). This is very useful
    for testing NonAffine behaviour with a simple Affine transform.

    Fr   c                    s   || _ tƒ j|i |¤Ž d S r;   )Ú
real_transr   r   )r   r£   r   r   r   r   r   r   Ú   s   zNonAffineForTest.__init__c                 C   ó   | j  |¡S r;   )r£   r"   )r   Úvaluesr   r   r   r#   Þ   ó   z%NonAffineForTest.transform_non_affinec                 C   r¤   r;   )r£   r   r   r   r   r   r    á   r¦   z*NonAffineForTest.transform_path_non_affine)r$   r%   r&   r'   r*   r)   r(   r   r#   r    r+   r   r   r   r   r¢   Ï   s    r¢   c                   @   s<   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ ZdS )ÚTestBasicTransformc                 C   sà   t jdd tjd ¡| _t jdd dd¡| _t jdd dd¡| _	t
t  ¡  dd¡d	d| _t
t  ¡  dd¡d
d| _t
t  ¡  dd¡dd| _| j| j| j  | j	 | _| j| j | j | j	 | _| j| j | j	 | _d S )NÚta1)Úshorthand_namer   Úta2r   r   Úta3rT   Útn1Útn2Útn3)r   r   Úrotater2   Úpir¨   r   rª   r   r«   r¢   r¬   r­   r®   Ústack1Ústack2Ústack2_subset©r   r   r   r   Úsetup_methodæ   s   ÿÿÿzTestBasicTransform.setup_methodc                 C   s4   | j jdksJ ‚| jjdksJ ‚| jjdksJ ‚d S )NrP   rW   )r±   Údepthr²   r³   r´   r   r   r   Útest_transform_depthÿ   s   z'TestBasicTransform.test_transform_depthc                 C   s¢   | j | j| j| j   | j }|| j| j| j  | j | j| j | j | j| j | jg}dd„ | ¡ D ƒ}t|ƒt|ƒks?J ‚t||ƒD ]
\}}||ksNJ ‚qDd S )Nc                 S   s   g | ]\}}|‘qS r   r   )Ú.0Ú_Úrhr   r   r   Ú
<listcomp>  s    zCTestBasicTransform.test_left_to_right_iteration.<locals>.<listcomp>)r¨   r¬   rª   r­   r«   Ú_iter_break_from_left_to_rightÚlenrš   )r   Ústack3Útarget_transformsÚrÚtarget_stackÚstackr   r   r   Útest_left_to_right_iteration  s   
üÿz/TestBasicTransform.test_left_to_right_iterationc                 C   sN  | j | j | jksJ ‚| j| j | jksJ ‚| j| j | j ¡ ks#J ‚| j| j jdks.J ‚t t¡ | j | j  W d   ƒ n1 sDw   Y  | j| j	| j
  }| j	| j
 }|| | jksaJ ‚|| j	 || j	 ¡  ksoJ ‚| j | j
 | j| j| j	  ks€J ‚| j| j
 | j| j | j	 ks‘J ‚| j	| j
 | j
 | j
 | j	| j
 ks¥J ‚d S )NrT   )r±   r³   r¨   r²   rŒ   r¶   ÚpytestÚraisesÚ
ValueErrorrª   r«   r¬   )r   Úaff1Úaff2r   r   r   Útest_transform_shortcuts  s    ÿ""
ÿz+TestBasicTransform.test_transform_shortcutsc                 C   s:  | j | j }| j | j }||ksJ ‚|| jksJ ‚| |¡s J ‚| | j¡s(J ‚| | j ¡r0J ‚| | j | j  ¡r;J ‚||ksAJ ‚| j | j¡sJJ ‚| j | j¡sSJ ‚| j | j¡s\J ‚| j | j¡seJ ‚| j | j¡rnJ ‚| j | j¡rwJ ‚| j | j | j ¡sƒJ ‚| j | j | j ¡sJ ‚| j | j| j  ¡r›J ‚d S r;   )rª   r¨   Úcontains_branchr±   r«   r²   r³   r¬   )r   Úr1Úr2r   r   r   Útest_contains_branch*  s$   z'TestBasicTransform.test_contains_branchc                 C   s  t jddgddgt jdgddggt jd}| j |¡}| j |¡}t jddgd	d
gt jt jgddggt jd}t jddgddgt jt jgddggt jd}t||ƒ t||ƒ t| j |¡|ƒ t| j 	¡  |¡|ƒ | j
| j  ¡ }| j 	¡  ¡ }t||ƒ | j 	¡  ¡ }t||ƒ d S )Nr   r   rS   rT   r~   )Údtyper‹   rp   g      3Àç      (@g      &@g      @g      "Àg      8@)r2   rt   ÚnanÚfloat64r±   r#   r"   r   Útransform_affiner™   rª   r«   rE   r   r²   )r   r‡   Úna_ptsÚall_ptsÚna_expectedÚall_expectedÚexpected_resultrž   r   r   r   Útest_affine_simplificationD  s8    ÿÿÿÿþ

ÿÿ
z-TestBasicTransform.test_affine_simplificationN)	r$   r%   r&   rµ   r·   rÃ   rÉ   rÍ   rØ   r   r   r   r   r§   å   s    r§   c                   @   s\   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ ZdS )ÚTestTransformPlotInterfacec                 C   sT   t  ¡ }|jg d¢g d¢|jd t|j ¡ t tj	tj	gtj	 tj	 gg¡ƒ d S )N©rm   ç333333ó?çš™™™™™é?©çÍÌÌÌÌÌì?ç      à?rÜ   r-   )
r/   r0   r1   r6   r   ru   rv   r2   rt   Úinf©r   r9   r   r   r   Útest_line_extent_axes_coordsk  s   
ÿÿz7TestTransformPlotInterface.test_line_extent_axes_coordsc                 C   sH   t  ¡ }|jg d¢g d¢|jd t|j ¡ t ddgddgg¡ƒ d S )NrÚ   rÝ   r-   rm   rß   rÛ   rÞ   )	r/   r0   r1   r4   r   ru   rv   r2   rt   rá   r   r   r   Útest_line_extent_data_coordss  s
   
ÿz7TestTransformPlotInterface.test_line_extent_data_coordsc                 C   s\   t  ¡ }t |j|j¡}|jg d¢g d¢|d t|j 	¡ t
 t
jdgt
j dgg¡ƒ d S )NrÚ   ©é#   éûÿÿÿé   r-   g      Àg     €A@)r/   r0   r   Úblended_transform_factoryr6   r4   r1   r   ru   rv   r2   rt   rà   ©r   r9   rœ   r   r   r   Ú!test_line_extent_compound_coords1z  s   ÿ

ÿÿz<TestTransformPlotInterface.test_line_extent_compound_coords1c                 C   sZ   t  ¡ }t ¡  d¡|j }|jg d¢g d¢|d t|j 	¡ t
 ddgddgg¡ƒ d S )	Nr   rÚ   rä   r-   r‹   ç      IÀrÏ   ç     àu@)r/   r0   r   r   r   r4   r1   r   ru   rv   r2   rt   ré   r   r   r   Ú)test_line_extent_predata_transform_coords…  s   
ÿzDTestTransformPlotInterface.test_line_extent_predata_transform_coordsc                 C   sj   t  ¡ }t |jt ¡  d¡|j ¡}|jg d¢g d¢|d t	|j
 ¡ t tjdgtj dgg¡ƒ d S )Nr   rÚ   rä   r-   rë   rì   )r/   r0   r   rè   r6   r   r   r4   r1   r   ru   rv   r2   rt   rà   ré   r   r   r   Ú!test_line_extent_compound_coords2  s   ÿ
ÿz<TestTransformPlotInterface.test_line_extent_compound_coords2c                 C   s`   t  ¡ }t ¡  dd¡}t jt d¡||j d t 	ddgddgg¡d }t
|j ¡ |ƒ d S )Nr   r-   ç        ç      "@©r/   r0   r   r   r   r1   r2   r3   r4   rt   r   ru   rv   ©r   r9   ÚoffsetÚexpected_data_limr   r   r   Útest_line_extents_affine—  s
   z3TestTransformPlotInterface.test_line_extents_affinec                 C   sx   t  ¡ }t ¡  dd¡}tt ¡  dd¡ƒ}t jt d¡|| |j	 d t 
ddgddgg¡d }t|j ¡ |ƒ d S )Nr   r-   rï   rð   rS   )r/   r0   r   r   r   r¢   r1   r2   r3   r4   rt   r   ru   rv   )r   r9   ró   Ú	na_offsetrô   r   r   r   Útest_line_extents_non_affinež  s   z7TestTransformPlotInterface.test_line_extents_non_affinec                 C   sœ   t  ¡ }t ¡  dd¡}tt ¡  dd¡ƒ}tddgddgddgddggƒ}tj||| |j	 d}| 
|¡ t ddgddgg¡d }t|j ¡ |ƒ d S )Nr   r   r-   rï   ç      $@rS   )r/   r0   r   r   r   r¢   r
   ÚmpatchesÚ	PathPatchr4   Ú	add_patchr2   rt   r   ru   rv   )r   r9   ró   rö   ÚpthÚpatchrô   r   r   r   Útest_pathc_extents_non_affine¦  s    ÿ
z8TestTransformPlotInterface.test_pathc_extents_non_affinec                 C   s„   t  ¡ }t ¡  dd¡}tddgddgddgddggƒ}tj|||j d}| 	|¡ t
 ddgddgg¡d }t|j ¡ |ƒ d S )Nr   r   r-   rï   rø   )r/   r0   r   r   r   r
   rù   rú   r4   rû   r2   rt   r   ru   rv   )r   r9   ró   rü   rý   rô   r   r   r   Útest_pathc_extents_affine±  s    
z4TestTransformPlotInterface.test_pathc_extents_affinec                 C   sh   t jdd}t ¡  dd¡}t jt d¡||j d t 	ddgddgg¡ddg }t
|j ¡ |ƒ d S )NÚpolar©Ú
projectionr   r   r-   rï   rð   rñ   rò   r   r   r   Ú*test_line_extents_for_non_affine_transDataº  s
   zETestTransformPlotInterface.test_line_extents_for_non_affine_transDataN)r$   r%   r&   râ   rã   rê   rí   rî   rõ   r÷   rþ   rÿ   r  r   r   r   r   rÙ   j  s    
	rÙ   c                 C   s   t | j|jƒ d S r;   )r   Úbounds)Úbbox1Úbbox2r   r   r   Úassert_bbox_eqÇ  s   r  c                  C   s0   t jjdddddd} |  ¡ }t|j| jƒ d S )Nrï   r‹   )Úminpos)r   r   Úfrom_extentsÚfrozenr   r  )Úbboxr
  r   r   r   Útest_bbox_frozen_copies_minposË  s   r  c                  C   sÀ   t jj} t jj}| ddddƒ}| ddddƒ}| ddddƒ}| ddddƒ}| ddddƒ}t|||ƒ|ƒ t|||ƒ| ddddƒƒ t|||ƒ|ƒ |||ƒd u sQJ ‚t|||ƒ| ddddƒƒ d S )Nr   rT   rß   rn   ç      è?g      @r   )r   r   r	  Úintersectionr  )Úbbox_from_extÚinterrË   rÌ   Úr3Úr4Úr5r   r   r   Útest_bbox_intersectionÑ  s   r  c                  C   s°   t  ddgddgg¡} t| tt| ƒdt jiƒƒ tt| ƒdtiƒ}| ¡ D ]\}}t| |ƒ|ks2J ‚q%d}tt	| |ƒdtiƒ}| ¡ D ]\}}tt	t| |ƒ|ƒƒ|ksUJ ‚qCd S )Nrß   r   r  r   z.1f)
r   r   r  ÚevalÚreprÚstrÚdictÚitemsÚgetattrÚformat)ÚbÚasdictÚkrh   Úfmtr   r   r   Útest_bbox_as_stringsç  s   ÿr   c                   C   s   t tjddjƒdksJ ‚d S )Nr   r  aØ  CompositeGenericTransform(
    CompositeGenericTransform(
        CompositeGenericTransform(
            TransformWrapper(
                BlendedAffine2D(
                    IdentityTransform(),
                    IdentityTransform())),
            CompositeAffine2D(
                Affine2D().scale(1.0),
                Affine2D().scale(1.0))),
        PolarTransform(
            PolarAxes(0.125,0.1;0.775x0.8),
            use_rmin=True,
            _apply_theta_transforms=False)),
    CompositeGenericTransform(
        CompositeGenericTransform(
            PolarAffine(
                TransformWrapper(
                    BlendedAffine2D(
                        IdentityTransform(),
                        IdentityTransform())),
                LockableBbox(
                    Bbox(x0=0.0, y0=0.0, x1=6.283185307179586, y1=1.0),
                    [[-- --]
                     [-- --]])),
            BboxTransformFrom(
                _WedgeBbox(
                    (0.5, 0.5),
                    TransformedBbox(
                        Bbox(x0=0.0, y0=0.0, x1=6.283185307179586, y1=1.0),
                        CompositeAffine2D(
                            Affine2D().scale(1.0),
                            Affine2D().scale(1.0))),
                    LockableBbox(
                        Bbox(x0=0.0, y0=0.0, x1=6.283185307179586, y1=1.0),
                        [[-- --]
                         [-- --]])))),
        BboxTransformTo(
            TransformedBbox(
                Bbox(x0=0.125, y0=0.09999999999999998, x1=0.9, y1=0.9),
                BboxTransformTo(
                    TransformedBbox(
                        Bbox(x0=0.0, y0=0.0, x1=8.0, y1=6.0),
                        Affine2D().scale(80.0))))))))r  r/   Úsubplotr4   r   r   r   r   Útest_str_transformó  s   r"  c                  C   s$   t  ¡ } |  d¡}|jdksJ ‚d S )N©rT   rT   )r   )r   r   rÒ   Úshape)rˆ   rÀ   r   r   r   Útest_transform_single_point%  s   
r%  c                  C   s&   t  ¡ \} }| d¡ |j d¡ d S )Nrs   r#  )r/   ÚsubplotsÚ
set_yscaler4   r"   )Úfigr9   r   r   r   Útest_log_transform+  s   
r)  c                  C   s@   t  ddgddgg¡} t  ddgdtjgg¡}|  |¡rJ ‚d S )Nr   rT   )r   r   r2   rÐ   Úoverlaps)Úar  r   r   r   Útest_nan_overlap3  s   r,  c                  C   sÚ   t  ¡ } t g d¢¡}t ddgddgddgg¡}|  ||¡}t||ƒ t t¡ |  ||dd…dd…f ¡ W d   ƒ n1 sBw   Y  t t¡ |  ||dd…d d …f ¡ W d   ƒ d S 1 sfw   Y  d S )N)rS   é-   r„   r   rT   r   )	r   r   r2   rt   Útransform_anglesr   rÄ   rÅ   rÆ   )rˆ   Úanglesr‡   Ú
new_anglesr   r   r   Útest_transform_angles9  s   
ÿ"ÿr1  c                  C   sF   t  ddg¡} dt jfddg}|D ]}t  tj|Ž ¡}t|| ƒ qd S )Ngü©ñÒMbP¿gü©ñÒMbP?r   ©r   r   )r   güó     )r2   rt   rÐ   r   Únonsingularr   )Úzero_expansionÚcasesr   Úoutr   r   r   Útest_nonsingularK  s   þr7  c                  C   sX  t  ¡ } t t¡ |  d¡ W d   ƒ n1 sw   Y  t t¡ |  dggg¡ W d   ƒ n1 s6w   Y  t t¡ |  g ¡ W d   ƒ n1 sPw   Y  t t¡ |  dg¡ W d   ƒ n1 skw   Y  t t¡ |  dgg¡ W d   ƒ n1 s‡w   Y  t t¡ |  g d¢g¡ W d   ƒ d S 1 s¥w   Y  d S )NrT   )rT   r   rW   )r   r   rÄ   rÅ   rÆ   r"   ÚRuntimeError©rˆ   r   r   r   Útest_invalid_argumentsT  s&   ÿÿÿÿÿ"ÿr:  c                  C   sÂ   g d¢} t | dd}t ¡ }t ||¡}t| ¡ j| ƒ dt d¡ }| 	tj
d ¡ t| ¡ jd||fdd| f| |fgd	d
 dgd |_t| ¡ jd||fdd| f| |fgd	d
 d S )N)r2  )rT   r   r#  )r   rT   T)ÚclosedrT   r   rP   r2  r   gVçž¯Ò<)Úatol)r
   r   r   ÚTransformedPathr   Úget_fully_transformed_pathÚverticesr2   Úsqrtr¯   r°   r‡   )r‡   r   rœ   Ú
trans_pathrÌ   r   r   r   Útest_transformed_pathi  s    
þ

þrB  c                  C   sl   t  ¡ } tjdddd| d}t  |¡}| ¡ j}|  d¡ t| ¡ j|d ƒ | 	d¡ t| ¡ j|ƒ d S )Nr2  rT   r-  é‡   r-   r   rß   )
r   r   rù   ÚWedgeÚTransformedPatchPathr>  r?  r   r   Ú
set_radius)rœ   rý   Útpatchr‡   r   r   r   Útest_transformed_patch_path  s   



rH  Úlocked_element©Úx0Úy0Úx1Úy1c                 C   sv  g d¢}|  | ¡ tj ¡ }tj|fi | di¤Ž}t|| ƒdks"J ‚t|d|  ƒdks-J ‚|D ]}t||ƒt||ƒks=J ‚q/| | ¡ d ¡ t|| ƒdksPJ ‚t|d|  ƒdks[J ‚|D ]}t||ƒt||ƒkskJ ‚q]t|d|  d ƒ t|d|  ƒd u sJ ‚t	 
| ¡ | ¡ k¡sŒJ ‚t|d|  dƒ t|| ƒdksJ ‚t|d|  ƒdks¨J ‚|D ]}t||ƒt||ƒks¸J ‚qªd S )NrJ  r   Úlocked_r   rW   )Úremover   r   ÚunitÚLockableBboxr  Ú
set_pointsrv   Úsetattrr2   Úall)rI  Úother_elementsÚorigÚlockedÚelemr   r   r   Útest_lockable_bbox  s,   

ÿrZ  c                  C   s²   t  ¡ } t  ¡ }| | }| ¡  t |¡}|js|jrJ ‚|  dd¡ |jr)|js+J ‚| ¡ |  ¡ k ¡ s7J ‚| ¡  t |¡}| dd¡ |jrKJ ‚| ¡ |  ¡ k ¡ sWJ ‚d S ©NrT   r   rW   rP   )r   r   rE   ÚcopyÚ_invalidr   rU  ©r+  r  ÚsÚs1Úb1r   r   r   Ú	test_copy²  s   


rb  c                  C   s¶   t  ¡ } t  ¡ }| | }| ¡  t |¡}|js|jrJ ‚|  dd¡ |jr)|jr+J ‚| ¡ t  ¡  ¡ k ¡ s9J ‚| ¡  t |¡}| dd¡ |jrMJ ‚| ¡ |  ¡ k ¡ sYJ ‚d S r[  )r   r   rE   r\  Údeepcopyr]  r   rU  r^  r   r   r   Útest_deepcopyÅ  s   


rd  c                  C   sR   t  t  ¡ ¡} tjtdd |  t d¡¡ W d   ƒ d S 1 s"w   Y  d S )Nz`The input and output dims of the new child \(1, 1\) do not match those of current child \(2, 2\)©Úmatchr   )	r   ÚTransformWrapperr   rÄ   rÅ   rÆ   Úsetr   r—   r9  r   r   r   Útest_transformwrapperØ  s   ÿ"ýri  Úpng)Ú
extensionsc              	   C   s    t j d¡ t jjdd}t  ddd¡}t| |gddgƒD ]0\}}| ¡ }|j||dd | |t  	|d	  d	 ¡t  
d	t j ¡ ¡ |j ¡  | d
¡ qd S )Ni!N,r   )Úsizeræ   rU   TF)rs   Údensityr   Úlinear)r2   ÚrandomÚseedÚnormalr[   rš   r&  Úhistr1   Úexpr@  r°   Úcanvasr5   r'  )Úfig_testÚfig_refÚsamplesre   r(  Ú	log_stater9   r   r   r   Útest_scale_swappingà  s   ,
ûry  c                   C   s~   t jtdd tjd dd W d   ƒ n1 sw   Y  t jtdd tjd dd W d   ƒ d S 1 s8w   Y  d S )NzZ'fontsize' is not a valid value for units; supported values are 'dots', 'points', 'inches're  Úfontsize)Úunitsz3For units of inches or points a fig kwarg is neededÚinches)rÄ   rÅ   rÆ   r   Úoffset_copyr   r   r   r   Útest_offset_copy_errorsî  s   ÿýÿ"þr~  c                  C   s’   t t ¡ tƒ  d¡ƒ} |  dd¡sJ ‚|  dd¡sJ ‚|  dd¡r#J ‚t t ¡ tƒ  dd¡ƒ} |  dd	¡s7J ‚|  dd	¡r?J ‚|  d
d
¡rGJ ‚d S )Né   rÜ   rß   gš™™™™™Ù¿g333333ë?rÞ   g      Ð?rq   rn   rm   )r	   r   rQ  r   Ú
rotate_degÚcontainsr   Úfully_contains)Úbbr   r   r   Útest_transformedbbox_containsù  s   r„  )?r\  Únumpyr2   Únumpy.testingr   r   r   r   rÄ   Ú
matplotlibr   Úmatplotlib.pyplotÚpyplotr/   Úmatplotlib.patchesÚpatchesrù   Úmatplotlib.transformsÚ
transformsr   r   r   r	   Úmatplotlib.pathr
   Úmatplotlib.testing.decoratorsr   r   r:   rG   ri   r{   r€   r‚   rŠ   r   r¡   r.   r¢   r§   rÙ   r  r  r  r   r"  r%  r)  r,  r1  r7  r:  rB  rH  ÚmarkÚparametrizerZ  rb  rd  ri  ry  r~  r„  r   r   r   r   Ú<module>   sb    &
ÿ
#% ]2	
!

