a
    ü
þdGt  ã                   @   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__© úi/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/matplotlib/tests/test_transforms.pyr      s    zEtest_non_affine_caching.<locals>.AssertingNonAffineTransform.__init__c                 S   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    s    ÿzVtest_non_affine_caching.<locals>.AssertingNonAffineTransform.transform_path_non_affinec                 S   s   | j rJ dƒ‚| j |¡S r   )r   r   Ú	transformr   r   r   r   Útransform_non_affine&   s    ÿ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   Ú	transAxesZ
invalidate)r*   Zmy_transÚaxr   r   r   Útest_non_affine_caching   s    
r6   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   Zscale_factorr   r   r   r   9   s    z6test_external_transform_api.<locals>.ScaledBy.__init__c                 S   s   t  ¡  | j¡|j S r7   )r   r   r   r8   r2   )r   r.   r   r   r   Ú_as_mpl_transform<   s    ÿz?test_external_transform_api.<locals>.ScaledBy._as_mpl_transformN)r"   r#   r$   r   r9   r   r   r   r   ÚScaledBy8   s   r:   r   r+   r   éd   )r-   r.   r/   r0   r1   Zset_xlimZset_ylimr   Zget_transformZ_aÚ
get_matrixr   r   r   )r:   r5   Úliner   r   r   Útest_external_transform_api7   s    ÿr>   Zpre_transform_dataTZmpl20gš™™™™™©?)Zremove_textÚstyleZ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    Z	linewidthé   )r-   r.   r   r   r   Úcontourfr0   r1   Zreshaper2   Ú
pcolormeshÚlinspaceZscatterÚsinÚcosZnewaxisZ
streamplotÚhypotZquiverZbarbs)r5   Ztimes10ÚxÚyÚuÚvr   r   r   Útest_pre_transform_plottingI   s,    "ýÿ&"
ÿ2rT   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é   rF   çÍÌÌÌÌÌ(@ç      )@çš™™™™™¹?r+   ç      ø?ç×£p=
×ó?ç       @ç      ô?)r-   r.   r0   ÚmeshgridrL   rJ   Úlogr   r   r   r2   Úarrayr   ÚdataLimÚ
get_points©r5   ÚxsZysÚexpectedr   r   r   Ú!test_contour_pre_transform_limitsn   s    $ÿ
ÿre   c                  C   s’   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 ©NrU   rF   rV   rW   éÿÿÿÿrX   r+   rY   rZ   r[   r\   )r-   r.   r0   r]   rL   Zpcolorr^   r   r   r   r2   r_   r   r`   ra   rb   r   r   r   Ú test_pcolor_pre_transform_limitsy   s    $$ÿ
ÿrh   c                  C   s’   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 rf   )r-   r.   r0   r]   rL   rK   r^   r   r   r   r2   r_   r   r`   ra   rb   r   r   r   Ú$test_pcolormesh_pre_transform_limits…   s    $$ÿ
ÿri   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   rF   rg   rG   r   éþÿÿÿrI   é<   rC   éP   rH   rA   )r0   r_   r   r   Zfrom_valuesr    r   )ÚpointsÚtÚactualrd   r   r   r   Útest_Affine2D_from_values‘   s8    
þ










rp   c                  C   sX   ddg} t  ¡ }t| | | ¡  | ¡¡ƒ | dd¡ ¡  t| | | ¡  | ¡¡ƒ d S )Nç      ð?)r   r   r   r    ÚinvertedÚ	translater<   )Zpointrn   r   r   r   Ú test_affine_inverted_invalidated¶   s
    rt   c                  C   sh   t  g d¢¡} t t ¡ t dd¡¡}| | ¡}|j| 	¡ ddd}t
|Ž \}}t|| jd d… ƒ d S )N))çš™™™™™É?éÿÿÿ)çš™™™™™Ù?rv   )rw   rF   )ru   rF   r   Úclip)r   r   r;   r;   F)rx   Zsimplifyrg   )r
   Z_create_closedr   ZBlendedGenericTransformr   r   ÚLogTransformr   Ziter_segmentsÚ
get_affineÚzipr   Úcodes)r   ÚtransZtpathÚresultZtpointsZ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 r7   )Ú
real_transr   r   )r   r   r   r   r   r   r   r   Ú   s    zNonAffineForTest.__init__c                 C   s   | j  |¡S r7   )r   r    )r   Úvaluesr   r   r   r!   Þ   s    z%NonAffineForTest.transform_non_affinec                 C   s   | j  |¡S r7   )r   r   r   r   r   r   r   á   s    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)Zshorthand_namer   Úta2r   r   Úta3rG   Útn1Útn2Útn3)r   r   Úrotater0   Úpir„   rs   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s0J ‚d S )NrC   rI   )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sˆJ ‚qˆd S )Nc                 S   s   g | ]\}}|‘qS r   r   )Ú.0Ú_Zrhr   r   r   Ú
<listcomp>  ó    zCTestBasicTransform.test_left_to_right_iteration.<locals>.<listcomp>)r„   r‡   r…   rˆ   r†   Z_iter_break_from_left_to_rightÚlenr{   )r   Zstack3Ztarget_transformsÚrZtarget_stackÚstackr   r   r   Útest_left_to_right_iteration  s    
üz/TestBasicTransform.test_left_to_right_iterationc                 C   sT  | j | j | jksJ ‚| j| j | jks,J ‚| j| j | j ¡ ksFJ ‚| j| j jdks\J ‚t t¡ | j | j  W d   ƒ n1 sˆ0    Y  | j| j	| j
  }| j	| j
 }|| | jksÂJ ‚|| j	 || j	 ¡  ksÞJ ‚| j | j
 | j| j| j	  ksJ ‚| j| j
 | j| j | j	 ks&J ‚| j	| j
 | j
 | j
 | j	| j
 ksPJ ‚d S )NrG   )rŒ   rŽ   r„   r   rr   r‘   ÚpytestÚraisesÚ
ValueErrorr…   r†   r‡   )r   Zaff1Zaff2r   r   r   Útest_transform_shortcuts  s    *$$
ÿz+TestBasicTransform.test_transform_shortcutsc                 C   s@  | j | j }| j | j }||ks$J ‚|| jks2J ‚| |¡s@J ‚| | j¡sPJ ‚| | j ¡r`J ‚| | j | j  ¡rvJ ‚||ks‚J ‚| j | j¡s”J ‚| j | j¡s¦J ‚| j | j¡s¸J ‚| j | j¡sÊJ ‚| j | j¡rÜJ ‚| j | j¡rîJ ‚| j | j | j ¡sJ ‚| j | j | j ¡s"J ‚| j | j| j  ¡r<J ‚d S r7   )r…   r„   Z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   rF   rG   rg   )Zdtyperq   r[   g      3Àç      (@g      &@g      @g      "Àg      8@)r0   r_   ÚnanÚfloat64rŒ   r!   r    r   Útransform_affinerz   r…   r†   r<   r   r   )r   rm   Zna_ptsZall_ptsZna_expectedZall_expectedZ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©rX   ç333333ó?çš™™™™™é?©çÍÌÌÌÌÌì?ç      à?rª   r+   )
r-   r.   r/   r4   r   r`   ra   r0   r_   Úinf©r   r5   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+   rX   r­   r©   r¬   )	r-   r.   r/   r2   r   r`   ra   r0   r_   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-   r.   r   Úblended_transform_factoryr4   r2   r/   r   r`   ra   r0   r_   r®   ©r   r5   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+   rq   ç      IÀr¢   ç     àu@)r-   r.   r   r   r   r2   r/   r   r`   ra   r0   r_   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-   r.   r   r¶   r4   r   r   r2   r/   r   r`   ra   r0   r_   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-   r.   r   r   rs   r/   r0   r1   r2   r_   r   r`   ra   ©r   r5   Ú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¾   rF   )r-   r.   r   r   rs   r€   r/   r0   r1   r2   r_   r   r`   ra   )r   r5   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½   ç      $@rF   )r-   r.   r   r   rs   r€   r
   ÚmpatchesÚ	PathPatchr2   Ú	add_patchr0   r_   r   r`   ra   )r   r5   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-   r.   r   r   rs   r
   rÇ   rÈ   r2   rÉ   r0   r_   r   r`   ra   )r   r5   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©Z
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 r7   )r   Zbounds)Zbbox1Z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½   rq   )Úminpos)r   r   Úfrom_extentsÚfrozenr   rÒ   )Z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 s¢J ‚t|||ƒ| ddddƒƒ d S )Nr   rG   r­   rY   ç      è?g      @r   )r   r   rÓ   ÚintersectionrÑ   )Zbbox_from_extÚinterrŸ   r    Zr3Zr4Z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sJJ ‚qJd}tt	| |ƒdtiƒ}| ¡ D ]$\}}tt	t| |ƒ|ƒƒ|ks†J ‚q†d S )Nr­   r   rÖ   r   z.1f)
r   r   rÑ   ÚevalÚreprÚstrÚdictÚitemsÚgetattrÚformat)ÚbZasdictÚkrS   Ú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-   Zsubplotr2   r   r   r   r   Útest_str_transformó  s    rå   c                  C   s$   t  ¡ } |  d¡}|jdks J ‚d S )N©rG   rG   )r   )r   r   r¥   Úshape)rn   r˜   r   r   r   Útest_transform_single_point%  s    
rè   c                  C   s&   t  ¡ \} }| d¡ |j d¡ d S )Nr^   ræ   )r-   ÚsubplotsÚ
set_yscaler2   r    )Úfigr5   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   rG   )r   r   r0   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 s„0    Y  t t¡, |  ||dd…d d …f ¡ W d   ƒ n1 sÊ0    Y  d S )N)rF   é-   rk   r   rG   r   )	r   r   r0   r_   Ztransform_anglesr   r›   rœ   r   )rn   Zanglesrm   Z
new_anglesr   r   r   Útest_transform_angles9  s    
:rñ   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üó     )r0   r_   r£   r   Znonsingularr   )Zzero_expansionZcasesr   Úoutr   r   r   Útest_nonsingularK  s
    rô   c                  C   sZ  t  ¡ } t t¡ |  d¡ W d   ƒ n1 s20    Y  t t¡  |  dggg¡ W d   ƒ n1 sl0    Y  t t¡ |  g ¡ W d   ƒ n1 s 0    Y  t t¡ |  dg¡ W d   ƒ n1 sÖ0    Y  t t¡ |  dgg¡ W d   ƒ n1 s0    Y  t t¡  |  g d¢g¡ W d   ƒ n1 sL0    Y  d S )NrG   )rG   r   rI   )r   r   r›   rœ   r   r    ÚRuntimeError©rn   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)rò   )rG   r   ræ   )r   rG   T)ÚclosedrG   r   rC   rò   r   gVçž¯Ò<)Zatol)r
   r   r   ZTransformedPathr   Úget_fully_transformed_pathÚverticesr0   ÚsqrtrŠ   r‹   rm   )rm   r   r}   Z
trans_pathr    r   r   r   Útest_transformed_pathi  s     
þ
þrü   c                  C   sl   t  ¡ } tjdddd| d}t  |¡}| ¡ j}|  d¡ t| ¡ j|d ƒ | 	d¡ t| ¡ j|ƒ d S )Nrò   rG   rð   é‡   r+   r   r­   )
r   r   rÇ   ZWedgeZTransformedPatchPathrù   rú   r   r   Z
set_radius)r}   rË   Ztpatchrm   r   r   r   Útest_transformed_patch_path  s    



rþ   Úlocked_element©Zx0Zy0Úx1Úy1c                 C   s€  g d¢}|  | ¡ tj ¡ }tj|fi | di¤Ž}t|| ƒdksDJ ‚t|d|  ƒdksZJ ‚|D ]}t||ƒt||ƒks^J ‚q^| | ¡ d ¡ t|| ƒdks J ‚t|d|  ƒdks¶J ‚|D ]}t||ƒt||ƒksºJ ‚qºt|d|  d ƒ t|d|  ƒd u sþJ ‚t	 
| ¡ | ¡ k¡sJ ‚t|d|  dƒ t|| ƒdks>J ‚t|d|  ƒdksVJ ‚|D ] }t||ƒt||ƒksZJ ‚qZd S )Nr   r   Zlocked_r   rI   )Úremover   r   ÚunitZLockableBboxrß   Z
set_pointsra   Úsetattrr0   Úall)rÿ   Zother_elementsÚorigÚlockedÚelemr   r   r   Útest_lockable_bbox  s*    

r
  c                  C   s²   t  ¡ } t  ¡ }| | }| ¡  t |¡}|js6|jr:J ‚|  dd¡ |jrR|jsVJ ‚| ¡ |  ¡ k ¡ snJ ‚| ¡  t |¡}| dd¡ |jr–J ‚| ¡ |  ¡ k ¡ s®J ‚d S ©NrG   r   rI   rC   )r   r   r<   ÚcopyÚ_invalidrs   r  ©rî   rá   ÚsÚs1Úb1r   r   r   Ú	test_copy²  s    


r  c                  C   s¶   t  ¡ } t  ¡ }| | }| ¡  t |¡}|js6|jr:J ‚|  dd¡ |jrR|jrVJ ‚| ¡ t  ¡  ¡ k ¡ srJ ‚| ¡  t |¡}| dd¡ |jršJ ‚| ¡ |  ¡ k ¡ s²J ‚d S r  )r   r   r<   r  Údeepcopyr  rs   r  r  r   r   r   Útest_deepcopyÅ  s    


r  c                  C   sP   t  t  ¡ ¡} tjtdd  |  t d¡¡ W d   ƒ n1 sB0    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   ZTransformWrapperr   r›   rœ   r   Úsetr   ry   rö   r   r   r   Útest_transformwrapperØ  s
    ÿr  Zpng)Ú
extensionsc              	   C   s    t j d¡ t jjdd}t  ddd¡}t| |gddgƒD ]`\}}| ¡ }|j||dd | |t  	|d	  d	 ¡t  
d	t j ¡ ¡ |j ¡  | d
¡ q:d S )Ni!N,r   )Úsizer´   rH   TF)r^   Zdensityr   Zlinear)r0   ÚrandomÚseedÚnormalrL   r{   ré   Úhistr/   Úexprû   r‹   Zcanvasr3   rê   )Zfig_testZfig_refZsamplesrP   rë   Z	log_stater5   r   r   r   Útest_scale_swappingà  s    ,
r   c                   C   s|   t jtdd tjd dd W d   ƒ n1 s20    Y  t jtdd tjd dd W d   ƒ n1 sn0    Y  d S )NzZ'fontsize' is not a valid value for units; supported values are 'dots', 'points', 'inches'r  Zfontsize)Zunitsz3For units of inches or points a fig kwarg is neededZinches)r›   rœ   r   r   Zoffset_copyr   r   r   r   Útest_offset_copy_errorsî  s    ÿ,ÿr!  c                  C   s’   t t ¡ tƒ  d¡ƒ} |  dd¡s&J ‚|  dd¡s6J ‚|  dd¡rFJ ‚t t ¡ tƒ  dd¡ƒ} |  dd	¡snJ ‚|  dd	¡r~J ‚|  d
d
¡rŽJ ‚d S )Né   rª   r­   gš™™™™™Ù¿g333333ë?r¬   g      Ð?r\   rY   rX   )r	   r   r  r   Z
rotate_degÚcontainsrs   Zfully_contains)Zbbr   r   r   Útest_transformedbbox_containsù  s    r$  )?r  Únumpyr0   Znumpy.testingr   r   r   r   r›   Z
matplotlibr   Zmatplotlib.pyplotZpyplotr-   Zmatplotlib.patchesZpatchesrÇ   Zmatplotlib.transformsZ
transformsr   r   r   r	   Zmatplotlib.pathr
   Zmatplotlib.testing.decoratorsr   r   r6   r>   rT   re   rh   ri   rp   rt   r   r,   r€   rƒ   r§   rÑ   rÕ   rÙ   rä   rå   rè   rì   rï   rñ   rô   r÷   rü   rþ   ÚmarkZparametrizer
  r  r  r  r   r!  r$  r   r   r   r   Ú<module>   s`   &
ÿ
#% ]2	
!

