OwlCyberSecurity - MANAGER
Edit File: test_streamplot.cpython-38.pyc
U ��]� � @ s$ d dl Z d dlZd dlZd dlmZ d dlmZ d dl m Z d dl m Z e jdkZdd� Zdd� Ze d gd d d �d d� �Ze dgdd d d�dd� �Ze dgddi�e�� d�d d d�dd� �Ze dger�dnd d d d�dd � �Ze d gd gd d d �d d!� �Ze d"gd gd d d �d#d$� �Zd%d&� ZdS )'� N)�assert_array_almost_equal)�image_comparisonZwin32c C sJ t jddd�ddd�f \} }d|d | }d| | d }|| ||fS )N���� y Y@���� � )�npZmgrid)�Y�X�U�V� r �B/usr/lib/python3/dist-packages/matplotlib/tests/test_streamplot.py�velocity_field s r c C sx t �ddd�} t �ddd�}t �| |�\}}d}t �|�| t �|�| }t �|�| t �|�| }| |||fS )Ng �g @�d g�������?)r �linspace�meshgrid�cos�sin)�x�yr r �ar r r r r �swirl_velocity_field s r Zstreamplot_startpointsTZmpl20)�baseline_images� remove_text�stylec C sn t � \} }}}t�| �� | �� d�}t�|�� |�� d�}t�||g�}tj| ||||d� t�||d� d S )N� )� start_points�ok) r r r �min�maxZ column_stack�plt� streamplotZplot)r r r r Zstart_xZstart_yr r r r �test_startpoints s r$ Zstreamplot_colormapg{�G�z�?)r Ztolr r c C s8 t � \} }}}tj| ||||ddtjjd� t�� d S )Ng333333�?r )�color�density� linewidth�cmap)r r" r# �cmZautumnZcolorbar)r r r r r r r � test_colormap* s �r* Zstreamplot_linewidthZaarch64g{�G�z�?� c C sV t � \} }}}t�||�}d| |�� }d}tj| |||d| d| gd|d� d S )N� g�������?g �?g �?�k)r& r% r' )r r Zhypotr! r" r# )r r r r ZspeedZlwZdfr r r �test_linewidth3 s �r. Zstreamplot_masks_and_nansc C s� t � \} }}}tj|jtd�}d|dd�dd�f< tj|d d�d d�f< tjj||d�}tjdd�� t j | ||||t j j d � W 5 Q R X d S ) N)Zdtyper �( �< � )�mask�ignore)Zinvalid)r% r( ) r r Zzeros�shape�bool�nanZmaZarrayZerrstater" r# r) ZBlues)r r r r r2 r r r �test_masks_and_nans? s r7 Zstreamplot_maxlengthZpng)r � extensionsr r c C s2 t � \} }}}tj| |||dddggddd� d S )Ng $@r+ � �?r )� maxlengthr r' r& �r r" r# �r r r r r r r �test_maxlengthL s �r= Zstreamplot_directionc C s4 t � \} }}}tj| |||ddddggddd� d S )NZbackwardr9 r+ r )Zintegration_directionr: r r' r& r; r< r r r �test_directionT s �r> c C s� t �� } t�ddd�}t�ddd�}t�||�\}}t�� �dd�| j }t j ||t� |�t� |�|d� t | j jd d d � d S ) N���r r1 ���� � � )Z transform)r1 � � � r )�decimal)r" Zaxesr r r � mtransformsZAffine2D� translateZ transDataZbarbsr r r ZdataLimZbounds)Zaxr r Ztransr r r �test_streamplot_limits] s �rJ )�sys�platformZnumpyr Z numpy.testingr Zmatplotlib.pyplotZpyplotr" Z matplotlib.testing.decoratorsr Zmatplotlib.transformsZ transformsrH Zon_winr r r$ r* �get�machiner. r7 r= r> rJ r r r r �<module> sZ � � � � � �