OwlCyberSecurity - MANAGER
Edit File: test_arrow_patches.cpython-38.pyc
U ��]8 � @ s8 d dl Z d dlZd dlmZ d dlmZ d dlmZ dd� Z edgd�dd� �Z ed gd gd �d d � �Z dd� Z edgdd gddi�e�� d�edd�d�dd� �Zedgdd gddi�e�� d�edd�d�d d � �Zed gdd gd d �d!d"� �Zed#gd gd$dd%�d&d'� �Zed(gd gd$dd%�d)d*� �Zd+d,� ZdS )-� N)�image_comparisonc C s* | j ddd| fddt|ddd�d� d S ) N� � �?�r r � �b�k)� arrowstyle�fc�ec)ZxyZxytext�sizeZ arrowprops)Zannotate�dict)�ax�t�r� r �E/usr/lib/python3/dist-packages/matplotlib/tests/test_arrow_patches.py� draw_arrow s ��r Zfancyarrow_test_image)�baseline_imagesc C s� dddddg} ddt j�� g}tjt|�t| �dtd d �d d �\}}t| �D ]@\}}t|�D ].\}}|||f }t|||� |j ddd � q\qLd S )N皙�����?�333333�?皙�����?皙�����?r ZfancyZsimpleFT)Zaspect)� g @)ZsqueezeZ subplot_kw�figsize)Z labelleftZ labelbottom) �mpatches� ArrowStyleZFancy�plt�subplots�lenr � enumerater Z tick_params) r r �figZaxesZi_rZr1Zi_t�t1r r r r �test_fancyarrow s � r# Zboxarrow_test_imageZpng)r � extensionsc C s� t j�� } t| �}d}|| d }tjd|d fd�}d}tt| ��D ]<\}}|jd|| | d | |d||j t |dd d �d � qFd S ) Ng333333�?r gUUUUUU@g �?�r g������5@�center�wr )Zboxstyler r )Zhar Z transformZbbox) r ZBoxStyle� get_stylesr r �figurer �sorted�textZ transFigurer )�styles�nZspacingZ figheightr! Zfontsize�i� stylenamer r r � test_boxarrow s �r0 c C sT t jdddd�} | �d�}|�ddg� |�ddg� |�tjdd d d d d �� | S )z� Convenience function that prepares and returns a FancyArrowPatch. It aims at being used to test that the size of the arrow head does not depend on the DPI value of the exported picture. NB: this function *is not* a test in itself! Zfancyarrow_dpi_cor_test)� � �2 )r �dpi�o r � )r r )皙�����?g333333�?r2 �->�d )ZposAZposBZlwr �mutation_scale)r r) Z add_subplot�set_xlim�set_ylim� add_patchr �FancyArrowPatch)Zfig2r r r r �!__prepare_fancyarrow_dpi_cor_test4 s �r? Zfancyarrow_dpi_cor_100dpiTZaarch64g{�G�z�?g r9 )r4 )r � remove_textr$ ZtolZ savefig_kwargc C s t � dS )a Check the export of a FancyArrowPatch @ 100 DPI. FancyArrowPatch is instantiated through a dedicated function because another similar test checks a similar export but with a different DPI value. Remark: test only a rasterized format. N�r? r r r r � test_fancyarrow_dpi_cor_100dpiF s rB Zfancyarrow_dpi_cor_200dpi�� c C s t � dS )z As test_fancyarrow_dpi_cor_100dpi, but exports @ 200 DPI. The relative size of the arrow head should be the same. NrA r r r r � test_fancyarrow_dpi_cor_200dpiV s rD Zfancyarrow_dash�default)r r@ r$ �stylec C s` ddl m} t�� \}}| ddddddd d d �}| dddd ddd d d �}|�|� |�|� d S )Nr )r> )r r r z-|>zangle3,angleA=0,angleB=90g $@� Zdashedr )r �connectionstyler: Z linewidthZ linestyleZcolorZangle3Zdotted)�matplotlib.patchesr> r r r= )r> r! r �eZe2r r r �test_fancyarrow_dashc s( �� rK Z arrow_stylesZmpl20)r r$ rF r@ c C sx t j�� } t| �}tjdd�\}}|�dd� |�d|� tt | ��D ],\}}t j d|fd|f|dd �}|� |� qFd S ) N�� � r% r r6 ���r r7 � )r r: ) r r r( r r r r; r< r r* r> r= �r, r- r! r r. r/ Zpatchr r r �test_arrow_styles} s �rR Zconnection_stylesc C s~ t j�� } t| �}tjdd�\}}|�dd� |�d|� tt | ��D ]2\}}t j d|fd|d fd |d d �}|� |� qFd S ) NrL r% r r6 rO r r7 r r8 rP )r rH r: ) r �ConnectionStyler( r r r r; r< r r* r> r= rQ r r r �test_connection_styles� s �rT c C sv t jjddd�} t jdd| d�}t�t�� t�� � |� W 5 Q R X t jjddd�}t jdd|d�}t�� � |� d S )N� rC )ZangleAZangleB)r r r )rH g�����h@) r rS ZAngle3r> �pytestZraises� ValueErrorr Zgcar= )Z conn_style_1Zp1Z conn_style_2Zp2r r r �test_invalid_intersection� s � �rX )rV �platformZmatplotlib.pyplotZpyplotr Z matplotlib.testing.decoratorsr rI Zpatchesr r r# r0 r? �get�machiner rB rD rK rR rT rX r r r r �<module> sR � � � � �