OwlCyberSecurity - MANAGER
Edit File: test_preprocess_data.cpython-38.pyc
U ��]+' � @ s� d dl Z d dlZd dlZd dlmZ eddgdd�d dd��ZegZdgZd d � Z ej j d eed �d d� �Z ej j d eed �dd� �Z ej j d eed �dd� �Zej j d eed �dd� �Zdd� Zdd� Zdd� Zdd � ZdS ) � N)�_preprocess_data�x�y�� replace_names� label_namer�xyzc C s dt |�t |�|||f S �Nz&x: %s, y: %s, ls: %s, w: %s, label: %s��list��axr r Zls�label�w� r �G/usr/lib/python3/dist-packages/matplotlib/tests/test_preprocess_data.py� plot_func s �r c C s dd� } dd� }dd� }dd� }t d d gd �| � t d d gd �|� t d d gd �|� t�t�� t d d d gd �|� W 5 Q R X t g d d�| � t g d d�|� t g d d�|� t g d d�|� t�t�� t d d�| � W 5 Q R X t�t�� t d d�|� W 5 Q R X d S )z-test decorator invocations -> no replacementsc S s d S �Nr )r r r r r r �func � z%test_compiletime_checks.<locals>.funcc W s d S r r )r r r �argsr r r � func_args r z*test_compiletime_checks.<locals>.func_argsc [ s d S r r )r r r �kwargsr r r � func_kwargs r z,test_compiletime_checks.<locals>.func_kwargsc _ s d S r r )r r r r r �func_no_ax_args r z0test_compiletime_checks.<locals>.func_no_ax_argsr r �r �zNr �r )r �pytest�raises�AssertionError)r r r r r r r �test_compiletime_checks s" r! r )Zidsc C s� | ddd�dkst �| dddd�dks*t �| ddddd�dksBt �| dddd d�d ksZt �| ddddd �dksrt �| dddd d �d ks�t �dS ) z$test without data -> no replacementsNr r z.x: ['x'], y: ['y'], ls: x, w: xyz, label: None)r r � )r z*x: ['x'], y: ['y'], ls: x, w: xyz, label: �textz.x: ['x'], y: ['y'], ls: x, w: xyz, label: text)r r r �r )r r r r � test_function_call_without_data; s$ � �����r% c C s� ddgddgdd�}| ddd |d �d ks,t �| ddd |d �d ksDt �| ddd d |d�dks^t �| ddd d|d�dksxt �| ddd d |d�dks�t �| ddd d|d�dks�t �dS )zCTest with dict data -> label comes from the value of 'x' parameter � � � � �NOT��a�br Nr, r- ��data�-x: [1, 2], y: [8, 9], ls: x, w: xyz, label: b�r r r/ r" �r r/ �,x: [1, 2], y: [8, 9], ls: x, w: xyz, label: r# �0x: [1, 2], y: [8, 9], ls: x, w: xyz, label: text�r r r r/ r$ �r r/ r r r �!test_function_call_with_dict_dataL s& ������r7 c C s� ddgdd�}| ddd|d�d ks&t �| ddd|d �d ks>t �| dddd |d �d ksXt �| dddd|d �dksrt �| dddd |d�d ks�t �| dddd|d�dks�t �dS )zItest for the case that one var is not in data -> half replaces, half keptr& r' r* )r, r Nr, r- r. z,x: [1, 2], y: ['b'], ls: x, w: xyz, label: br1 r" r2 z+x: [1, 2], y: ['b'], ls: x, w: xyz, label: r# z/x: [1, 2], y: ['b'], ls: x, w: xyz, label: textr5 r$ r6 r r r �-test_function_call_with_dict_data_not_in_data^ s& ������r8 c C s� |� tjddgtjd�tjddgtjd�ddgd��}| dd d |d �d ksNt�| dd d |d �d ksft�| dd d d|d�dks�t�| dd d d|d�dks�t�| dd d d|d�dks�t�| dd d d|d�dks�t�dS )z@test with pandas dataframe -> label comes from data["col"].name r& r' )Zdtyper( r) r* r+ Nr, r- r. r0 r1 r" r2 r3 r# r4 r5 )Z DataFrame�npZarrayZint32r )r Zpdr/ r r r �#test_function_call_with_pandas_datap s, �������r: c C s� ddgddgdd�} t dd�dd d ��}|d ddd | d�dksBt�|d ddd | d�dks\t�|d ddd d| d�dksxt�|d ddd d| d�dks�t�|d ddd d| d�dks�t�|d ddd d| d�dks�t�d S )zDTest without a "replace_names" argument, all vars should be replacedr& r' r( r) r )r, r- r r r r Nr* c S s dt |�t |�|||f S r r r r r r �func_replace_all� s �z8test_function_call_replace_all.<locals>.func_replace_allr, r- )r r/ r0 )r r r r/ r" )r r r/ r3 r# r4 )r r r r r/ )r Nr* �r r )r/ r; r r r � test_function_call_replace_all� s0 ���������r= c C s� t ddgdd�ddd��} dd gd d gd d �}| ddd|d�dksFt�| ddd|d�dks^t�| dddd|d�dksxt�| dddd|d�dks�t�dS )z;Test with "label_namer=None" -> no label replacement at allr r Nr r c S s dt |�t |�|||f S r r r r r r � func_no_label� s �z1test_no_label_replacements.<locals>.func_no_labelr& r' r( r) r* r+ r, r- r. z0x: [1, 2], y: [8, 9], ls: x, w: xyz, label: Noner1 r" r2 r3 r# r4 )r Nr r< )r> r/ r r r �test_no_label_replacements� s ����r? c C s\ t ddgdd�ddd��} ddgdd gd d �}t�t�� | d d d dd|d� W 5 Q R X d S )Nr r r r& c S s d S r r )r r r r r r r r � s z/test_more_args_than_pos_parameter.<locals>.funcr' r( r) r* r+ r, r- r r. )r& )r r r � TypeErrorr6 r r r �!test_more_args_than_pos_parameter� s rA c C sZ t � dd� �} t�d| j�s t�t�d| j�r2t�t�d| j�rDt�t g d�ddd��} t�d| j�rjt�t�d| j�r|t�t�d| j�r�t�t dgd�dd d��} t�d| j�r�t�t�d | j�s�t�t�d| j�r�t�t d dgd�dd d��} t�d | j��st�t�d| j��st�t�d| j��s.t�t�d| j��rBt�t�d| j��rVt�d S )Nc _ s dS �zFuncy does nothingNr )r r r r r r �funcy� s z&test_docstring_addition.<locals>.funcyz,.*All positional and all keyword arguments\.z .*All positional arguments\.z,.*All arguments with the following names: .*r c S s dS rB r �r r r r �barr r r rC � s rE c S s dS rB r rD r r r rC � s z1.*All arguments with the following names: 'bar'\.r c S s dS rB r rD r r r rC � s z6.*All arguments with the following names: '.*', '.*'\.z.*'x'.*z .*'bar'.*)N)N)N)r �re�search�__doc__r )rC r r r �test_docstring_addition� s\ ��� ��� ��� � � �rI )r Nr )rF Znumpyr9 r Z matplotlibr r Z all_funcsZ all_func_idsr! ZmarkZ parametrizer% r7 r8 r: r= r? rA rI r r r r �<module> s( !