OwlCyberSecurity - MANAGER
Edit File: test_quiver.cpython-38.pyc
U ��] � @ sj d dl Z d dlZd dlZd dlZd dlmZ d dlm Z dd� Z dd� Z dd � Z d d � Z d d � Ze dgdgd�dd� �Ze dgdgd�dd� �Ze dgdgdd�dd� �Zdd � Ze d gdgdd�d d � �Ze d gdgdd�d!d"� �Ze d#gdgdd�d$d%� �Ze d&gdgdd�d'd(� �Ze d)gddgd*�d+d,� �Zd-d.� Zd/d0� Ze d1gdgdd�d2d3� �Zd4d5� ZdS )6� N)�pyplot)�image_comparisonc K sX t �t �ddt j d�t �ddt j d��\}}t �|�}t �|�}| j||f|�}|S )Nr � � )�np�meshgrid�arange�pi�cos�sin�quiver)�ax�kw�X�Y�U�V�Q� r �>/usr/lib/python3/dist-packages/matplotlib/tests/test_quiver.py� draw_quiver s � r c C s: t �� \} }t|�}|j}|�� ~t�|�dks6t�d S )Nr )�plt�subplotsr r �remove�sys� getrefcount�AssertionError)�figr r ZttXr r r �test_quiver_memory_leak s r c C s` t �� \} }t|�}|j|dddddddid�}t�|�d ksBt�|�� t�|�dks\t�d S ) N� �?�q= ףp�?r �$2 \frac{m}{s}$�W�weight�bold��labelpos�fontproperties� )r r r � quiverkeyr r r r �r r r �qkr r r �test_quiver_key_memory_leak s �r, c C s� t �� \} }t�t�d�t�d��\}}t�|� }}tj�dd�d d }tj dd��$}|j |||||d� | j � � W 5 Q R X t |�dks�t�d S ) N� � r � T��record��anglesr )r r r r r � ones_like�random�rand�warnings�catch_warningsr �canvas�draw�lenr )r r r r r r �phi�wr r r �test_no_warnings, s r> c C s� t �� \} }t�t�d�t�d��\}}t�|�t�|� }}tjdd��"}|j ||ddd� | j � � W 5 Q R X t |�dks�t �d S )Nr. Tr0 r )Z headlengthZheadaxislength)r r r r r r r r7 r8 r r9 r: r; r )r r r r r r r= r r r �test_zero_headlength9 s r? Zquiver_animated_test_imageZpng)�baseline_images� extensionsc C s8 t �� \} }t|dd�}|j|ddddddd id �}d S ) NT)Zanimatedr r r r! r" r# r$ r% )r r r r) r* r r r �test_quiver_animateE s �rB Zquiver_with_key_test_imagec C sD t �� \} }|�d� t|�}|j|dddddddd d d �d � }d S ) N皙�����?r gffffff�?r z"$2\, \mathrm{m}\, \mathrm{s}^{-1}$i����Zfigurer" r$ Zlarge)r# �size)�angleZ coordinatesr& r' )r r �marginsr r) r* r r r �test_quiver_with_keyR s ��rG Zquiver_single_test_imageT)r@ rA � remove_textc C s2 t �� \} }|�d� |�dgdgdgdg� d S )NrC r r )r r rF r �r r r r r �test_quiver_singlec s rJ c C sf t �� \} }tt�dg�t�dg�d�}|�dgdg|d |d �}d|d d< |jd dksbt�d S )N皙�����?g @)�u�vr rL rM r )r r �dictr �arrayr r r )r r �uvZq0r r r �test_quiver_copyl s rQ Zquiver_key_pivotc C s� t �� \} }tjddtj d�ddtj d�f \}}|�t�|�t�|��}|�dd� |� dd� |j |dddddd � |j |dddd d d � |j |dddd d d � |j |dddd d d � d S ) Nr r y $@���� r r �N)r& �E�Sr" ) r r r Zmgridr r r r �set_xlim�set_ylimr) )r r rL rM �qr r r �test_quiver_key_pivott s * rZ Z quiver_key_xyc C s� t �d�} t �d�}| t jd }t �d| �}|j}|j}t�d�\}}t |d�D ]j\}} |� dd� |� dd� |j | |||d d d dd | d � } t d d�D ] \} } |j | | dd| ddd� q�qTd S )N� � y �?r )rP �xy���g������ɿ皙�����?Zmiddler] g�������?)�pivotZunits�width�scale� scale_unitsr3 )r_ r 皙�����?)r �- �Z rd r � �b)r r r rE ZlabelZcolor)r r Zzerosr Zexp�real�imagr r �ziprW rX r r) ) r r r3 rP r r r Zaxsr Z angle_strrY �xrE r r r �test_quiver_key_xy� s& �rm Zbarbs_test_imagec C sn t �ddd�} t �| | �\}}d| d| }}t�� \}}|j||||t �||�ddtdddd �d d � d S ) N���� � TF� �?r_ �333333�?�Z emptybarbZspacingZheightZviridis)� fill_empty�rounding�sizesZcmap)r �linspacer r r �barbsZhypotrN �rl r r r r r r r r r � test_barbs� s �rz Zbarbs_pivot_test_imagec C sv t �ddd�} t �| | �\}}d| d| }}t�� \}}|j||||dddtddd d �d � |j||d d d� d S )Nrn ro rp TF�333333�?rq r_ rr rs )rt ru r` rv �1 Zblack)�s�c)r rw r r r rx rN Zscatterry r r r �test_barbs_pivot� s �r Zbarbs_test_flip)rH rA c C sj t �ddd�} t �| | �\}}d| d| }}t�� \}}|j||||dddtddd d �|d k d � d S )z'Test barbs with an array for flip_barb.rn ro rp TFr{ rq r_ rr rs r )rt ru r` rv Z flip_barbN)r rw r r r rx rN ry r r r �test_barbs_flip� s �r� c C s� t �d�} t �d�}t j�dt �d� �}t j�dt �|� �}t jj|d< t jj|d< t�� \}}t � t �� |� | |||� W 5 Q R X dS )z<Test error handling when given differing sized masked arraysr( g .@)r\ r N) r r ZmarO Zonesr4 Zmaskedr r �pytestZraises� ValueErrorrx )rl �yrL rM r r r r r �test_bad_masked_sizes� s r� c C sf t �� \} }t�t�d�t�d��\}}t�|� }}tj�dd�d d }|j|||||dd� d S )Nr- r. r r/ r] )r3 rc ) r r r r r r4 r5 r6 r )r r r r r r r< r r r �test_angles_and_scale� s r� Z quiver_xyc C sR t jtdd�d�\} }|jdddddddd� |�dd� |�dd� |�� d S ) NZequal)Zaspect)Z subplot_kwr r r] )r3 rc rb rK )r r rN r rW rX ZgridrI r r r �test_quiver_xy� s r� c C s� t �� \} }t�t�d�t�d��\}}t�|� } }}|j|||||d�}|�|dddd�}| j� � t |j �dks|t �d S )Nr r2 r ZLabel) r r r r r r4 r r) r9 r: r; Zvertsr ) r r r r r r r3 rY r+ r r r �test_quiverkey_angles� s r� ) r7 Znumpyr r� r Z matplotlibr r Z matplotlib.testing.decoratorsr r r r, r> r? rB rG rJ rQ rZ rm rz r r� r� r� r� r� r r r r �<module> sl � � � � � � � �