U
    Âmœd·p  ã                   @   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 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+d/d0„ Z,d1d2„ Z-d3d4„ Z.d5d6„ Z/d7d8„ Z0d9d:„ Z1d;d<„ Z2ej3 4d=d>d?d@dAg¡dBdC„ ƒZ5dDdE„ Z6dFdG„ Z7dHdI„ Z8edJgdKdLdM„ ƒZ9dS )Né    N)Úassert_allcloseÚassert_almost_equalÚassert_array_equalÚassert_array_almost_equal)Úscale)Ú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||Ž d| _t ¡  dd¡| _d S )NFé
   )ÚsuperÚ__init__Úraise_on_transformÚmtransformsÚAffine2Dr   Úunderlying_transform)ÚselfÚargsÚkwargs©Ú	__class__© úY/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/matplotlib/tests/test_transforms.pyr      s    zEtest_non_affine_caching.<locals>.AssertingNonAffineTransform.__init__c                 S   s   | j rtdƒ‚| j |¡S ©Nz3Invalidated affine part of transform unnecessarily.)r   ÚAssertionErrorr   Ú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tdƒ‚| j |¡S r   )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    
r5   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   8   s    z6test_external_transform_api.<locals>.ScaledBy.__init__c                 S   s   t  ¡  | j¡|j S r6   )r   r   r   r7   r1   )r   r-   r   r   r   Ú_as_mpl_transform;   s    ÿz?test_external_transform_api.<locals>.ScaledBy._as_mpl_transformN)r!   r"   r#   r   r8   r   r   r   r   ÚScaledBy7   s   r9   r   r*   r   éd   )r,   r-   r.   r/   r0   Zset_xlimZset_ylimr   Zget_transformZ_aÚ
get_matrixr   r   r   )r9   r4   Úliner   r   r   Útest_external_transform_api6   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   Úcontourfr/   r0   Zreshaper1   Ú
pcolormeshÚlinspaceZscatterÚsinÚcosZnewaxisZ
streamplotÚhypotZquiverZbarbs)r4   Ztimes10ÚxÚyÚuÚvr   r   r   Útest_pre_transform_plottingH   s,    "ýÿ&"
ÿ2rS   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é   rE   çÍÌÌÌÌÌ(@ç      )@çš™™™™™¹?r*   ç      ø?ç×£p=
×ó?ç       @ç      ô?)r,   r-   r/   ÚmeshgridrK   rI   Úlogr   r   r   r1   Úarrayr   ÚdataLimÚ
get_points©r4   ZxsZysÚexpectedr   r   r   Ú!test_contour_pre_transform_limitsm   s    $ÿ
ÿrc   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 ©NrT   rE   rU   rV   éÿÿÿÿrW   r*   rX   rY   rZ   r[   )r,   r-   r/   r\   rK   Zpcolorr]   r   r   r   r1   r^   r   r_   r`   ra   r   r   r   Ú test_pcolor_pre_transform_limitsx   s    $$ÿ
ÿrf   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 rd   )r,   r-   r/   r\   rK   rJ   r]   r   r   r   r1   r^   r   r_   r`   ra   r   r   r   Ú$test_pcolormesh_pre_transform_limits„   s    $$ÿ
ÿrg   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   rE   re   rF   r
   éþÿÿÿrH   é<   rB   éP   rG   r@   )r/   r^   r   r   Zfrom_valuesr   r   )ÚpointsÚtÚactualrb   r   r   r   Útest_Affine2D_from_values   s8    
þ










rn   c                  C   sX   ddg} t  ¡ }t| | | ¡  | ¡¡ƒ | dd¡ ¡  t| | | ¡  | ¡¡ƒ d S )Nç      ð?)r   r   r   r   ÚinvertedÚ	translater;   )Úpointrl   r   r   r   Ú test_affine_inverted_invalidatedµ   s
    rs   c                  C   sl   t  ddddg¡} t t ¡ t dd¡¡}| | ¡}|j| 	¡ ddd	}t
|Ž \}}t|| jd d
… ƒ d S )N)çš™™™™™É?éÿÿÿ)çš™™™™™Ù?ru   )rv   rE   )rt   rE   r   Úclip)r   r   r:   r:   F)rw   Úsimplifyre   )r   Z_create_closedr   ZBlendedGenericTransformr   r   ÚLogTransformr   Ú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||Ž d S r6   )Ú
real_transr   r   )r   r‚   r   r   r   r   r   r   Ù   s    zNonAffineForTest.__init__c                 C   s   | j  |¡S r6   )r‚   r   )r   Úvaluesr   r   r   r    Ý   s    z%NonAffineForTest.transform_non_affinec                 C   s   | j  |¡S r6   )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   Úta3rF   Útn1Útn2Útn3)r   r   Úrotater/   Úpir…   rq   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t‚| jjdks t‚| jjdks0t‚d S )NrB   rH   )r   Údepthr   rŽ   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~t‚t||ƒD ]\}}||ksˆt‚qˆd S )Nc                 S   s   g | ]\}}|‘qS r   r   )Ú.0Ú_Zrhr   r   r   Ú
<listcomp>  s     zCTestBasicTransform.test_left_to_right_iteration.<locals>.<listcomp>)	r…   rˆ   r†   r‰   r‡   Z_iter_break_from_left_to_rightÚlenr   r|   )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   s>  | j | j | jkst‚| j| j | jks,t‚| j| j | j ¡ ksFt‚| j| j jdks\t‚t t	¡ | j | j  W 5 Q R X | j| j
| j  }| j
| j }|| | jks®t‚|| j
 || j
 ¡  ksÊt‚| j | j | j| j| j
  ksìt‚| j| j | j| j | j
 kst‚| j
| j | j | j | j
| j ks:t‚d S )NrF   )r   r   r…   r   rŽ   rp   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$t‚|| jks2t‚| |¡s@t‚| | j¡sPt‚| | j ¡r`t‚| | j | j  ¡rvt‚||ks‚t‚| j | j¡s”t‚| j | j¡s¦t‚| j | j¡s¸t‚| j | j¡sÊt‚| j | j¡rÜt‚| j | j¡rît‚| j | j | j ¡st‚| j | j | j ¡s"t‚| j | j| j  ¡r<t‚d S r6   )	r†   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   rE   rF   re   )Zdtypero   rZ   g      3Àç      (@g      &@g      @g      "Àg      8@)r/   r^   ÚnanZfloat64r   r    r   r   Útransform_affiner{   r†   r‡   r;   r   rŽ   )r   rk   Zna_ptsZall_ptsZna_expectedZall_expectedZexpected_resultr   r   r   r   Útest_affine_simplificationC  s<     ÿ
 ÿÿ
 ÿþ

ÿÿ
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   sX   t  ¡ }|jdddgdddg|jd t|j ¡ t tj	tj	gtj	 tj	 gg¡ƒ d S ©NrW   ç333333ó?çš™™™™™é?gÍÌÌÌÌÌì?ç      à?r*   )
r,   r-   r.   r3   r   r_   r`   r/   r^   Úinf©r   r4   r   r   r   Útest_line_extent_axes_coordsj  s    
ÿÿz7TestTransformPlotInterface.test_line_extent_axes_coordsc                 C   sL   t  ¡ }|jdddgdddg|jd t|j ¡ t ddgddgg¡ƒ d S r§   )	r,   r-   r.   r1   r   r_   r`   r/   r^   r¬   r   r   r   Útest_line_extent_data_coordsr  s
    
ÿz7TestTransformPlotInterface.test_line_extent_data_coordsc                 C   s`   t  ¡ }t |j|j¡}|jdddgdddg|d t|j 	¡ t
 t
jdgt
j d	gg¡ƒ d S )
NrW   r¨   r©   é#   éûÿÿÿé   r*   g      Àg     €A@)r,   r-   r   Úblended_transform_factoryr3   r1   r.   r   r_   r`   r/   r^   r«   ©r   r4   r~   r   r   r   Ú!test_line_extent_compound_coords1y  s    ÿ

ÿÿz<TestTransformPlotInterface.test_line_extent_compound_coords1c                 C   s^   t  ¡ }t ¡  d¡|j }|jdddgdddg|d t|j 	¡ t
 d	d
gddgg¡ƒ d S )Nr   rW   r¨   r©   r¯   r°   r±   r*   ro   ç      IÀr¢   ç     àu@)r,   r-   r   r   r   r1   r.   r   r_   r`   r/   r^   r³   r   r   r   Ú)test_line_extent_predata_transform_coords„  s    
ÿzDTestTransformPlotInterface.test_line_extent_predata_transform_coordsc                 C   sn   t  ¡ }t |jt ¡  d¡|j ¡}|jdddgdddg|d t	|j
 ¡ t tjd	gtj d
gg¡ƒ d S )Nr   rW   r¨   r©   r¯   r°   r±   r*   rµ   r¶   )r,   r-   r   r²   r3   r   r   r1   r.   r   r_   r`   r/   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   rq   r.   r/   r0   r1   r^   r   r_   r`   ©r   r4   Ú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º   rE   )r,   r-   r   r   rq   r   r.   r/   r0   r1   r^   r   r_   r`   )r   r4   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¹   ç      $@rE   )r,   r-   r   r   rq   r   r   ÚmpatchesÚ	PathPatchr1   Ú	add_patchr/   r^   r   r_   r`   )r   r4   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   rq   r   rÃ   rÄ   r1   rÅ   r/   r^   r   r_   r`   )r   r4   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¦   i  s   
	r¦   c                 C   s   t | j|jƒ d S r6   )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¹   ro   )Úminpos)r   ÚBboxÚ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 ks¢t‚t|||ƒ| ddddƒƒ d S )Nr   rF   rª   rX   ç      è?g      @r
   )r   rÏ   rÐ   ÚintersectionrÍ   r   )Zbbox_from_extZ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sJt	‚qJd}tt
| |ƒdtiƒ}| ¡ D ]$\}}tt
t| |ƒ|ƒƒ|ks†t	‚q†d S )Nrª   r   rÓ   rÏ   z.1f)r   rÏ   rÍ   ÚevalÚreprÚstrÚdictÚitemsÚgetattrr   Úformat)ÚbÚasdictÚkrR   Úfmtr   r   r   Útest_bbox_as_stringsæ  s    rá   c                   C   s   t tjddjƒdkst‚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subplotr1   r   r   r   r   r   Útest_str_transformò  s    râ   c                  C   s$   t  ¡ } |  d¡}|jdks t‚d S )N©rF   rF   )r
   )r   r   r¤   Úshaper   )rl   r˜   r   r   r   Útest_transform_single_point$  s    
rå   c                  C   s&   t  ¡ \} }| d¡ |j d¡ d S )Nr]   rã   )r,   ÚsubplotsÚ
set_yscaler1   r   )Úfigr4   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<t‚d S )Nr   rF   )r   rÏ   r/   r£   Úoverlapsr   )ÚarÝ   r   r   r   Útest_nan_overlap2  s    rì   c               	   C   s²   t  ¡ } t dddg¡}t ddgddgddgg¡}|  ||¡}t||ƒ t t¡" |  ||dd…dd…f ¡ W 5 Q R X t t¡" |  ||dd…d d …f ¡ W 5 Q R X d S )NrE   é-   ri   r   rF   r
   )	r   r   r/   r^   Ztransform_anglesr   r›   rœ   r   )rl   Zanglesrk   Z
new_anglesr   r   r   Útest_transform_angles8  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üó     )r/   r^   r£   r   Znonsingularr   )Zzero_expansionZcasesr   Úoutr   r   r   Útest_nonsingularJ  s
    rñ   c               	   C   sà   t  ¡ } t t¡ |  d¡ W 5 Q R X t t¡ |  dggg¡ W 5 Q R X t t¡ |  g ¡ W 5 Q R X t t¡ |  dg¡ W 5 Q R X t t¡ |  dgg¡ W 5 Q R X t t¡ |  dddgg¡ W 5 Q R X d S )NrF   r
   rH   )r   r   r›   rœ   r   r   ÚRuntimeError©rl   r   r   r   Útest_invalid_argumentsS  s    rô   c                  C   sÆ   ddddg} 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 )Nrï   )rF   r   rã   )r   rF   T)ÚclosedrF   r
   rB   r   gVçž¯Ò<)Zatol)r   r   r   ZTransformedPathr   Úget_fully_transformed_pathÚverticesr/   Úsqrtr‹   rŒ   rk   )rk   r   r~   Z
trans_pathr    r   r   r   Útest_transformed_pathh  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ï   rF   rí   é‡   r*   r
   rª   )
r   r   rÃ   ZWedgeZTransformedPatchPathrö   r÷   r   r   Z
set_radius)r~   rÇ   Ztpatchrk   r   r   r   Útest_transformed_patch_path~  s    



rû   Úlocked_elementÚx0Úy0Úx1Úy1c                 C   s€  ddddg}|  | ¡ tj ¡ }tj|f| diŽ}t|| ƒdksDt‚t|d|  ƒdksZt‚|D ]}t||ƒt||ƒks^t‚q^| | ¡ d ¡ t|| ƒdks t‚t|d|  ƒdks¶t‚|D ]}t||ƒt||ƒksºt‚qºt	|d|  d ƒ t|d|  ƒd ksþt‚t
 | ¡ | ¡ k¡st‚t	|d|  dƒ t|| ƒdks>t‚t|d|  ƒdksVt‚|D ] }t||ƒt||ƒksZt‚qZd S )	Nrý   rþ   rÿ   r   r
   Zlocked_r   rH   )Úremover   rÏ   ÚunitZLockableBboxrÛ   r   Z
set_pointsr`   Úsetattrr/   Ú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:t‚|  dd¡ |jrR|jsVt‚| ¡ |  ¡ k ¡ snt‚| ¡  t |¡}| dd¡ |jr–t‚| ¡ |  ¡ k ¡ s®t‚d S ©NrF   r
   rH   rB   )r   r   r;   ÚcopyÚ_invalidr   rq   r  ©rë   rÝ   ÚsÚs1Úb1r   r   r   Ú	test_copy±  s    


r  c                  C   s¶   t  ¡ } t  ¡ }| | }| ¡  t |¡}|js6|jr:t‚|  dd¡ |jrR|jrVt‚| ¡ t  ¡  ¡ k ¡ srt‚| ¡  t |¡}| dd¡ |jršt‚| ¡ |  ¡ k ¡ s²t‚d S r	  )	r   r   r;   r
  Údeepcopyr  r   rq   r  r  r   r   r   Útest_deepcopyÄ  s    


r  c               	   C   s<   t  t  ¡ ¡} tjtdd |  t d¡¡ W 5 Q R X 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°   rG   TF)r]   Zdensityr
   Zlinear)r/   ÚrandomÚseedÚnormalrK   r|   ræ   Úhistr.   Úexprø   rŒ   Zcanvasr2   rç   )Zfig_testZfig_refZsamplesrO   rè   Z	log_stater4   r   r   r   Útest_scale_swappingß  s    ,
r  ):r
  Únumpyr/   Znumpy.testingr   r   r   r   r›   Z
matplotlibr   Zmatplotlib.pyplotZpyplotr,   Zmatplotlib.patchesZpatchesrÃ   Zmatplotlib.transformsZ
transformsr   Zmatplotlib.pathr   Zmatplotlib.testing.decoratorsr   r	   r5   r=   rS   rc   rf   rg   rn   rs   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   Ú<module>   sZ   &
ÿ
#% ]2	
!
