U
    \	¸aµ  ã                   @   s–   d dl Zd dlZd dlmZ d dlmZ d dlm	Z	m
Z
 dd„ Ze
dgƒdd„ ƒZe
d	gƒd
d„ ƒZe	dgddd„ ƒZe
dgƒdd„ ƒZdd„ ZdS )é    N)ÚSpines)Úcheck_figures_equalÚimage_comparisonc               	   C   s  G dd„ dƒ} | ƒ | ƒ | ƒ | ƒ dœ}t f |Ž}|d |d ksBt‚|j|d ksTt‚|ddg  d¡ |jjdksvt‚|jjdks†t‚|jjdks–t‚|jjdks¦t‚|dd…  d¡ td	d
„ | 	¡ D ƒƒsÒt‚t
jtdd |d  W 5 Q R X t
jtdd |ddddg  W 5 Q R X t
jtdd |d  W 5 Q R X t
jtdd |dd…  W 5 Q R X t
jtdd |dd…  W 5 Q R X dS )z)Test Spines and SpinesProxy in isolation.c                   @   s   e Zd Zdd„ Zdd„ ZdS )z#test_spine_class.<locals>.SpineMockc                 S   s
   d | _ d S ©N©Úval)Úself© r	   ú@/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/tests/test_spines.pyÚ__init__   s    z,test_spine_class.<locals>.SpineMock.__init__c                 S   s
   || _ d S r   r   )r   r   r	   r	   r
   Úset_val   s    z+test_spine_class.<locals>.SpineMock.set_valN)Ú__name__Ú
__module__Ú__qualname__r   r   r	   r	   r	   r
   Ú	SpineMock   s   r   )ÚleftÚrightÚtopÚbottomr   r   ÚxNÚyc                 s   s   | ]}|j d kV  qdS )r   Nr   )Ú.0Úspiner	   r	   r
   Ú	<genexpr>$   s     z#test_spine_class.<locals>.<genexpr>Zfoo)Úmatchzfoo, barÚbarzsingle list)r   r   zSpines does not support slicingr   )r   ÚAssertionErrorr   r   r   r   r   r   ÚallÚvaluesÚpytestZraisesÚKeyErrorÚ
ValueError)r   Zspines_dictÚspinesr	   r	   r
   Útest_spine_class	   s4    ü
r#   Zspines_axes_positionsc                  C   s¢   t  ¡ } t ddtj d¡}dt |¡ }|  ddd¡}| d¡ | ||¡ |j	j
 d¡ |j d¡ |j	j d¡ |j d	¡ |j	j d
¡ |j	j d
¡ d S )Nr   é   éd   é   zcentered spines)Úaxesçš™™™™™¹?r   )r'   g      Ð?r   Únone)ÚpltÚfigureÚnpZlinspaceÚpiÚsinÚadd_subplotÚ	set_titleÚplotr"   r   Úset_positionÚyaxisÚset_ticks_positionr   Úxaxisr   Z	set_colorr   )Úfigr   r   Úaxr	   r	   r
   Útest_spines_axes_positions2   s    
r8   Zspines_data_positionsc                  C   sd   t  ¡ \} }|jj d¡ |jj d¡ |jj d¡ |jj d¡ | ddg¡ | 	ddg¡ d S )N)Údatag      ø¿)r9   g      à?)r9   g      à¿Zzeroéþÿÿÿr$   )
r*   Úsubplotsr"   r   r2   r   r   r   Zset_xlimZset_ylim©r6   r7   r	   r	   r
   Útest_spines_data_positionsC   s    r=   Zpng)Ú
extensionsc                 C   s–   t j d¡ |  ¡ }|jddd |jj d¡ |jj d¡ |jj	 d¡ |j
dd	d
d | ¡ }|jddd |jj	 d¡ |j
dd	d	dd d S )NÚdefaultÚlog)r(   r&   )ZxscaleZxlim)r9   r&   r$   )r9   r(   r   FÚin)ÚaxisÚ	labelleftÚ	directionT)rB   rC   r   r   )r*   ÚstyleZuser/   Úsetr"   r   r2   Zset_linewidthr   Ztick_params)Zfig_testZfig_refr7   r	   r	   r
   Ú#test_spine_nonlinear_data_positionsN   s    rG   Zspines_capstylec                  C   s2   t jddd t  ¡ \} }| g ¡ | g ¡ d S )Nr'   é   )Z	linewidth)r*   Úrcr;   Z
set_xticksZ
set_yticksr<   r	   r	   r
   Útest_spines_capstylec   s    
rJ   c                  C   s,  t  ¡ \} }t jddd | t d¡¡ |j d¡ |jj	 
d¡ |jj d¡ | d¡ |j d¡ |jj 
d¡ |jj d¡ | d	¡ |j g ¡ |j g ¡ t  ¡  |jj	}| ¡  | ¡ ¡ ¡ }|jj ¡ d
 |jk sêtdƒ‚|jj}| ¡  | ¡ ¡ ¡ }|jj ¡ d |jk s(tdƒ‚d S )Ng333333Ó?)r   r   é
   r   )Zoutwardé   Fzy labelr   zx labelr   z"Y-Axis label not left of the spiner&   z X-Axis label not below the spine)r*   r;   Zsubplots_adjustr1   r,   Zaranger3   r4   r"   r   r2   r   Zset_visibleZ
set_ylabelr5   r   r   Z
set_xlabelZ	set_ticksZdrawZget_transformZtransform_pathÚget_pathZget_extentsÚlabelÚget_positionZxminr   Zymin)r6   r7   r   Z	spinebboxr	   r	   r
   Útest_label_without_ticksl   s8    

ÿÿÿÿrP   )Znumpyr,   r   Zmatplotlib.pyplotZpyplotr*   Zmatplotlib.spinesr   Zmatplotlib.testing.decoratorsr   r   r#   r8   r=   rG   rJ   rP   r	   r	   r	   r
   Ú<module>   s   )





