OwlCyberSecurity - MANAGER
Edit File: test_skew.cpython-38.pyc
U ��]� � @ s� d Z ddlmZ ddlZddlmZ ddlmZ ddl m Z ddl m Z ddl mZ ddlmZ ddlmZ ddlmZ G dd� dej�ZG d d � d ej�ZG d d � d ej �Z G d d� de �Z ee � edgdd�dd� �Z edgdd�dd� �Z dS )z) Testing that skewed axes properly work. � )� ExitStackN)�image_comparison)�Axes)�register_projectionc s$ e Zd Z� fdd�Zdd� Z� ZS )� SkewXTickc s� t � ��}| j| j| j| j| jfD ]}|�|j|�� � q t � | j j | � � �}t � | j j| � � �}| j�| j�� or|� | j�| j�� o�|� | j�| j�� o�|� | j�| j�� o�|� tt| ��|� W 5 Q R X d S �N)r ZgridlineZ tick1lineZ tick2lineZlabel1Zlabel2�callbackZ set_visibleZ get_visible� transformsZinterval_contains�axes� lower_xlimZget_loc� upper_xlim�superr �draw)�selfZrenderer�stackZartistZ needs_lowerZ needs_upper�� __class__� �</usr/lib/python3/dist-packages/matplotlib/tests/test_skew.pyr s6 � � � � � � �zSkewXTick.drawc C s | j j�� S r )r �xaxis�get_view_interval�r r r r r ) s zSkewXTick.get_view_interval)�__name__� __module__� __qualname__r r � __classcell__r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )� SkewXAxisc C s t | jd d|d�S )N� )�major)r r )r r r r r � _get_tick0 s zSkewXAxis._get_tickc C s | j jd | j jd fS )Nr � )r r r r r r r r 3 s zSkewXAxis.get_view_intervalN)r r r r r r r r r r / s r c @ s e Zd Zdd� ZdS )� SkewSpinec C s@ | j j}| jdkr(| jj|d d �df< n| jj|d d �df< d S )N�topr )�_pathZverticesZ spine_typer r r �r Zptsr r r �_adjust_location; s zSkewSpine._adjust_locationN)r r r r% r r r r r! : s r! c @ s@ e Zd ZdZdd� Zdd� Zdd� Zedd � �Zed d � �Z d S ) � SkewXAxes�skewxc C sb t | �| _| jd �| j� | jd �| j� t�| �| _| jd �| j� | jd �| j� d S )Nr" �bottom�left�right)r r �spinesZ register_axis�maxisZYAxisZyaxisr r r r � _init_axisL s zSkewXAxes._init_axisc C s8 t �| d�tj�| d�tj�| d�tj�| d�d�}|S )Nr" r( r) r* )r" r( r) r* )r! Z linear_spine�mspines�Spine)r r+ r r r �_gen_axes_spinesU s �zSkewXAxes._gen_axes_spinesc C sn d}t �| � | j| jt�� �|d� | _| j| j | _ t� | j| j t� � �t�� �|d� | j | _ dS )z� This is called once when the plot is created to set up all the transforms for the data, text and grids. � r N) r �_set_lim_and_transformsZ transScaleZ transLimitsr �Affine2D�skew_deg�transDataToAxesZ transAxes� transDataZblended_transform_factoryZIdentityTransformZ_xaxis_transform)r Zrotr r r r2 \ s �� ���z!SkewXAxes._set_lim_and_transformsc C s | j jjS r )r ZviewLimZ intervalxr r r r r y s zSkewXAxes.lower_xlimc C s, ddgddgg}| j �� �|�d d �df S )Ng g �?r )r5 �inverted� transformr$ r r r r } s zSkewXAxes.upper_xlimN) r r r �namer- r0 r2 �propertyr r r r r r r& F s r& Z skew_axesT)Zbaseline_imagesZ remove_textc C sN t �� } | jddddd�}|�dd� |�dd� |�d� |jddd � d S ) Nr r' )Z projectioni����2 Tr �b)Zcolor)�pltZfigureZ add_subplot�set_xlim�set_ylimZgridZaxvline)Zfig�axr r r � test_set_line_coll_dash_image� s rA Z skew_rectsc C s� t jdddddd�\} }|j}ttjddddd gd d ��}|d �dd g� |d �dd g� |d jd dd � t ||�D ]f\}\}}d| d| }}t � � � ||�}|� d�||�� |�tjddgd d ||j ddd�� qzt jdddddd� d S )N� T)� rC )ZsharexZshareyZfigsize������r r � � )�repeatZequal)Zshare�- z Skew of {0} in X and {1} in Yg �?Zcoral)r8 ZalphaZ facecolorg{�G�z�?g�G�z��?)Zwspacer) r* r( r" )r= ZsubplotsZflat�list� itertools�productr> r? Z set_aspect�zipr r3 r4 Z set_title�formatZ add_patch�mpatchZ Rectangler6 Zsubplots_adjust) Zfixr Z rotationsr@ ZxrotsZyrotsZxdegZydeg�tr r r �test_skew_rectangle� s � rQ )!�__doc__� contextlibr rK Zmatplotlib.pyplotZpyplotr= Z matplotlib.testing.decoratorsr Zmatplotlib.axesr Zmatplotlib.transformsr Zmatplotlib.axisZaxisr, Zmatplotlib.spinesr+ r. Zmatplotlib.patchesZpatchesrO Zmatplotlib.projectionsr ZXTickr ZXAxisr r/ r! r& rA rQ r r r r �<module> s& ?