OwlCyberSecurity - MANAGER
Edit File: test_animation.cpython-38.pyc
U ��]�% � @ s� d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlmZ d dlZd dl Z d dl m Z d dl m Z G dd� de j�Zdd� Zd d � Zd d � Ze j�d �G dd� de��Zdddddddddg Zedd� eD �7 Zej�de�d d � �Zd d � Zd d!� Zejj e j� e j d" � d#d$�ej�d%d&d'g�d(d)� ��Z ejj e j� e j d" � d#d$�ej�d%d&ej d'ejj!d*�g�d+d,� ��Z"ejj e j#d-kd.d$�d/d0� �Z$ej�d1ej d2d3d4� d5d6�ej d7d8d4� d9d6�g�d:d;� �Z%dS )<� N��Path)�pyplot)� animationc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �NullMovieWritera� A minimal MovieWriter. It doesn't actually write anything. It just saves the arguments that were given to the setup() and grab_frame() methods as attributes, and counts how many times grab_frame() is called. This class doesn't have an __init__ method with the appropriate signature, and it doesn't define an isAvailable() method, so it cannot be added to the 'writers' registry. c G s" || _ || _|| _|| _d| _d S )Nr )�fig�outfile�dpi�args�_count)�selfr r r r � r �A/usr/lib/python3/dist-packages/matplotlib/tests/test_animation.py�setup s zNullMovieWriter.setupc K s || _ | jd7 _d S )N� )�savefig_kwargsr )r r r r r � grab_frame# s zNullMovieWriter.grab_framec C s d S �Nr �r r r r �finish' s zNullMovieWriter.finishN)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r c s< t �� \}}|�g �\� dd� }� fdd�}tj||f| �S )Nc S s d S r r r r r r �init/ s z make_animation.<locals>.initc s � � ddgd| g� � fS )Nr r ��set_data)�i��liner r �animate2 s z make_animation.<locals>.animate)�plt�subplots�plotr � FuncAnimation)�kwargsr �axr r r r r �make_animation+ s r' c C s� d} t | d�}d}d}tdd�}t� }|j||||d� |jt�d�ksLt�|j|ksZt�|j |ksht�|j d ksvt�|j |ks�t�|j | ks�t�d S ) N� ��frames� unused.null�2 r )Zfoo)r �writerr r r ) r' �dictr �saver r! �figure�AssertionErrorr r r r r )Z num_frames�anim�filenamer r r- r r r �test_null_movie_writer9 s �r4 c C s^ G dd� dt j�} t�� }d}d}d}d}dg}| ||||�}|�||� |j|jksZt�d S )Nc @ s e Zd Zdd� ZdS )z7test_movie_writer_dpi_default.<locals>.DummyMovieWriterc S s d S r r r r r r �_runQ s z<test_movie_writer_dpi_default.<locals>.DummyMovieWriter._runN)r r r r5 r r r r �DummyMovieWriterP s r6 r+ r( Zunusedr )r Z MovieWriterr! r0 r r r1 )r6 r r3 �fps�codec�bitrate� extra_argsr- r r r � test_movie_writer_dpi_defaultO s r; �nullc @ s" e Zd Zddd�Zedd� �ZdS )�RegisteredNullMovieWriterNc C s d S r r )r r7 r8 r9 r: Zmetadatar r r �__init__k s z"RegisteredNullMovieWriter.__init__c C s dS )NTr )�clsr r r � isAvailableo s z%RegisteredNullMovieWriter.isAvailable)NNNNN)r r r r> � classmethodr@ r r r r r= b s � r= )�ffmpeg� movie.mp4)Z ffmpeg_filerC )ZavconvrC )Z avconv_filerC )Z imagemagick� movie.gif)Zimagemagick_filerD )�pillowrD )Zhtmlz movie.html)r< z movie.nullc C s g | ]\}}|t |�f�qS r r )�.0r- �outputr r r � <listcomp> s rH zwriter, outputc s |dkrt �d� z |�� W n tk r2 Y nX tj�|�sNt �d| � t� � \}}|� g g �\� |� dd� |� dd� d }d }|dkr�|� d � d }d }� fd d �}� fdd�}| �� �T tj|||dd�} z| j|d|d||d� W n tk �r t �d� Y nX W 5 Q R X d S )NrE ZPILz(writer '%s' not available on this systemr � ���r rB )g33333�%@g�Q� �k"@g Y@Zh264c s � � g g � � fS r r r r r r r � s z+test_save_animation_smoketest.<locals>.initc s. t �ddd�}t �|| �}� �||� � fS )Nr rI �d )�np�linspaceZsinr )r �x�yr r r r � s z.test_save_animation_smoketest.<locals>.animater( )� init_funcr* � i� )r7 r- r9 r r8 zIThere can be errors in the numpy import stack, see issues #1891 and #2679)�pytestZ importorskipZ_init_from_registry�AttributeErrorr �writers� is_available�skipr! r" r# Zset_xlimZset_ylimZset_size_inches�as_cwdr$ r/ �UnicodeDecodeError�xfail) �tmpdirr- rG r r& r r8 r r r2 r r r � test_save_animation_smoketest� s8 � r[ c C s" t ttd��d�jdt� d� d S )Nr( r) r+ �r- )r' �iter�ranger/ r r r r r �test_no_length_frames� s �r_ c C s� t jd } ttjj�dks t�tj�� tjjr4t�dt jd< tjjsJt�tj�� tjjr`t�tj� d�rpt�t j dkr~dnd}|t jd< tjjs�t�tj�� tjjr�t�tj� d�s�t�| t jd< d S )Nzanimation.ffmpeg_pathr Znot_available_ever_xxxxrB Zwin32�true�where) �mpl�rcParams�lenr rT � _registeredr1 �list�_dirtyrU �sys�platform)Z ffmpeg_path�binr r r �test_movie_writer_registry� s rk zanimation.writerz animation writer not installed)�reason� method_nameZto_html5_videoZ to_jshtmlc C s� |� d� |�� �2 t�ddi�� ttdd�| �� W 5 Q R X W 5 Q R X t|j�dks\t�|j\}|j dkrx|j dks|t�d S )NZWARNINGzanimation.embed_limitg���ư>r r) zmatplotlib.animation) Z set_levelrW rb Z rc_context�getattrr' rd Zrecordsr1 �nameZ levelname)rm ZcaplogrZ �recordr r r �test_embed_limit� s & �rq )Zmarksc C sF |� � �4 ttdd�| �� ttt|���� �g ks8t�W 5 Q R X d S )Nr r) )rW rn r' rf r �strZiterdirr1 )rm rZ r r r �test_cleanup_temporaries� s rs �posixzrequires a POSIX OSc C s� z�| �� �r |�ddtjd � tt| �d�}|� d� t� t|�d� t j�� t � t j�� t� �d� W 5 Q R X W 5 Q R X W 5 t j�� X dS )a! Test that we correctly raise a CalledProcessError when ffmpeg fails. To do so, mock ffmpeg using a simple executable shell script that succeeds when called with no arguments (so that it gets registered by `isAvailable`), but fails otherwise, and add it to the $PATH. �PATHz.:rB z#!/bin/sh [[ $@ -eq 0 ]] i� z test.mpegN)r rT Zreset_available_writersrW Zsetenv�os�environr rr Z write_text�chmodrR Zraises� subprocessZCalledProcessErrorr' r/ )rZ Z monkeypatchZexe_pathr r r �test_failing_ffmpeg� s $rz z&cache_frame_data, weakref_assertion_fnFc C s | d kS r r ��refr r r �<lambda> � r} Z cache_frame_data_is_disabled)�idTc C s | d k S r r r{ r r r r} r~ Zcache_frame_data_is_enabledc s� t �� \}}|�g g �\�G dd� dt�� �fdd�}�fdd�}g �� �fdd�}tj||||| d �}t� }|jd |d � t��d ks�t ��D ]} || � �s�t �q�d S ) Nc @ s e Zd ZdS )z0test_funcanimation_holding_frames.<locals>.FrameN)r r r r r r r �Frame s r� c s � � g g � � fS r r r r r r r s z/test_funcanimation_holding_frames.<locals>.initc s � � | d | d � � fS )NrN rO r )�framer r r r s z2test_funcanimation_holding_frames.<locals>.animatec 3 sN t d�D ]@} t�ddd�}tj�d�}� ||d�}��t�|�� |V qd S )Nr( r rI rK )rN rO )r^ rL rM ZrandomZrand�append�weakrefr| )�_rN rO r� )r� �frames_generatedr r �frames_generator s z;test_funcanimation_holding_frames.<locals>.frames_generator)rP r* �cache_frame_datar+ r\ r( ) r! r" r# r. r r$ r r/ rd r1 ) r� Zweakref_assertion_fnr r& r r r� r2 r- �fr )r� r� r r �!test_funcanimation_holding_frames s �r� )&rv �pathlibr ry rh r� ZnumpyrL rR Z matplotlibrb r r! r ZAbstractMovieWriterr r' r4 r; rT �registerr= Z WRITER_OUTPUTZmarkZ parametrizer[ r_ rk ZskipifrU rc rq ZparamrY rs ro rz r� r r r r �<module> s� � � 0� ���� � ��