OwlCyberSecurity - MANAGER
Edit File: test_category.cpython-38.pyc
U ��]?( � @ s� d Z ddlZddlZddlmZ ddlmZ ddl m Z e j j ZG dd� de�ZG dd� de�ZG dd � d e�Zejd d � �ZejejejgZd d dgZG dd� de�ZG dd� de�Zdd� Z G dd� de�Z G dd� de�Z G dd� de�Z dS )z#Catch all for categorical functions� N)�Axesc @ s� e Zd Zddgdgffddgdgffdddd d d gdd d ddgffgZee� \ZZejj deed�dd� �Z dd� Z dde j fdddgfddd gfgZee� \ZZejj d eed�d d � �Zejj d eed�d d � �Zd!S )"� TestUnitData�single� hello worldr �unicode� Здравствуйте мир�mixed�Aznp.nan�Bz3.14� мир� � � � z data, locs��idsc C s: t �|�}t|j�� �|ks t�t|j�� �|ks6t�d S �N)�cat�UnitData�list�_mapping�keys�AssertionError�values)�self�data�locs�unit� r �@/usr/lib/python3/dist-packages/matplotlib/tests/test_category.py� test_unit s zTestUnitData.test_unitc C s� ddg}ddg}dddg}ddddg}ddddg}t �|�}t|j�� �|ksRt�t|j�� �|ksht�|�|� t|j�� �|ks�t�t|j�� �|ks�t�d S ) N�a�dr r �b�er r )r r r r r r r �update)r r r Z data_updateZ unique_dataZ updated_locsr r r r � test_update s zTestUnitData.test_update)Znumber� ��Q� @�nanr r' � z mixed type�fdatac C s$ t �t�� t�|� W 5 Q R X d S r )�pytest�raises� TypeErrorr r )r r* r r r �test_non_string_fails/ s z"TestUnitData.test_non_string_failsc C s, t �� }t�t�� |�|� W 5 Q R X d S r )r r r+ r, r- r% )r r* Zunitdatar r r � test_non_string_update_fails4 s z)TestUnitData.test_non_string_update_failsN)�__name__� __module__� __qualname__� test_cases�zipr r r+ �mark� parametrizer r& �npr( �failing_test_cases�fidsr* r. r/ r r r r r s& �� � r c @ s e Zd Zdd� ZdS )�FakeAxisc C s || _ d S r )�units)r r; r r r �__init__< s zFakeAxis.__init__N)r0 r1 r2 r< r r r r r: ; s r: c @ s2 e Zd ZdZddgfddgfdddd gfd d d gfd ddddddddddddddd d d d d d!d"d#d$d%d&d'gfgZee� \ZZd(d)dej gfd*d+d,gfgZ ee � \Z Z e jd-d.�d/d0� �Ze jjd1eed2�d3d4� �Ze jjd5d6d7gddgd2�d8d9� �Zd:d;� Zd<d=� Ze jjd>e e d2�d?d@� �ZdAdB� ZdCdD� ZdES )F�TestStrCategoryConverterz�Based on the pandas conversion and factorization tests: ref: /pandas/tseries/tests/test_converter.py /pandas/tests/test_algos.py:TestFactorize r r �asciir r r! r# �czinteger string�1�2zsingle + values>10r r �C�D�E�F�G�H�I�J�K�L�M�N�O�P�Q�R�S�T�U�V�W�X�Y�Zr r' �string integer�42�* T)Zautousec C s$ t �� | _t �� | _t| j�| _d S r )r ZStrCategoryConverter�ccr r r: �ax)r Zrequestr r r � mock_axisV s z"TestStrCategoryConverter.mock_axis�valsr c C s, t j�| j�|| jj| j�tt|��� d S r ) r7 �testing�assert_allcloser] �convertr^ r; �range�len)r r` r r r � test_convert] s � �z%TestStrCategoryConverter.test_convert�value�hir c C s | j �|| j| j�dks t�d S )Nr )r] rc r r^ r )r rg r r r �test_convert_one_stringc s z0TestStrCategoryConverter.test_convert_one_stringc C s. | j �d| j| j�}tj�|t�dg�� d S )Ng )r] rc r r^ r7 ra rb �array)r �actualr r r �test_convert_one_numberg s z0TestStrCategoryConverter.test_convert_one_numberc C sF t jdddgtd�}| j�|| j| j�}t j�|t �dddg�� d S )Nr r r )Zdtypeg �?g @g @) r7 rj �floatr] rc r r^ ra rb )r r rk r r r �test_convert_float_arrayk s z1TestStrCategoryConverter.test_convert_float_array�fvalsc C s. t �t�� | j�|| j| j� W 5 Q R X d S r )r+ r, r- r] rc r r^ )r ro r r r �test_convert_failp s z*TestStrCategoryConverter.test_convert_failc C s: | j �| j| j�}t|jtj�s$t�t|j tj �s6t�d S r ) r] Zaxisinfor r^ � isinstanceZmajlocr �StrCategoryLocatorr Zmajfmt�StrCategoryFormatter)r �axisr r r � test_axisinfou s z&TestStrCategoryConverter.test_axisinfoc C s" t | j�dg| j�tj�s t�d S )Nr! )rq r] Z default_unitsr^ r r r �r r r r �test_default_unitsz s z+TestStrCategoryConverter.test_default_unitsN)r0 r1 r2 �__doc__r3 r4 r r r7 �infr8 r9 �fvaluesr+ �fixturer_ r5 r6 rf ri rl rn rp ru rw r r r r r= @ sX �� � r= c C s t �� �� S r )�pltZfigureZsubplotsr r r r r^ ~ s r^ �scatter�plot�barc @ s. e Zd Zdd� Zejjdeed�dd� �Z dS )�TestStrCategoryLocatorc C sT ddddddddd d d g }t �d d � |D ��}t �|j�}tj�|�d d �|� d S )Nr r r r r � � � � � � c S s g | ] }t |��qS r )�str)�.0�jr r r � <listcomp>� s zBTestStrCategoryLocator.test_StrCategoryLocator.<locals>.<listcomp>)r r rr r r7 ra �assert_array_equalZ tick_values)r r r �ticksr r r �test_StrCategoryLocator� s z.TestStrCategoryLocator.test_StrCategoryLocator�plotterr c C s. |� dddg� tj�|jj�� td�� d S )Nr! r# r? r )r~ r7 ra r� �yaxis�majorZlocatorrd �r r^ r� r r r �test_StrCategoryLocatorPlot� s z2TestStrCategoryLocator.test_StrCategoryLocatorPlotN) r0 r1 r2 r� r+ r5 r6 � PLOT_LIST�PLOT_IDSr� r r r r r� � s r� c @ sx e Zd ZddddgfdddgfgZee� \ZZejj deed �d d � �Z ejj deed �ejj d e e d �d d� ��Z dS )�TestStrCategoryFormatterr> �hello�worldrh r � Здравствуйте� привет�ydatar c C sB t �|�}t �|j�}t|�D ] \}}|||�t|�ks t�q d S r )r r rs r � enumerate�_to_strr )r r^ r� r �labels�ir" r r r �test_StrCategoryFormatter� s z2TestStrCategoryFormatter.test_StrCategoryFormatterr� c C s~ ||t t|��|� t|�D ]$\}}|jj�||�t|�ks t�q |jj�|d |d �dksbt�|jj�dd �dkszt�d S )Nr � r )rd re r� r� r� � formatterr� r )r r^ r� r� r� r" r r r � test_StrCategoryFormatterPlot� s z6TestStrCategoryFormatter.test_StrCategoryFormatterPlotN)r0 r1 r2 r3 r4 r Zcasesr+ r5 r6 r� r� r� r� r r r r r� � s � r� c s� t tt|���}tj�� �� |� � fdd�|D �}|dd� |D �ksJt�t � jj � � �dd� |D �kslt�t � jj � � �|ks�t�d S )Nc s g | ]}� j �||��qS r )r� r� )r� r� �rt r r r� � s z axis_test.<locals>.<listcomp>c S s g | ] }t |��qS r )r� �r� �lr r r r� � s c S s g | ]}|�qS r r r� r r r r� � s ) r rd re r7 ra r� Zget_majorticklocsr r; r r r )rt r� r� Z graph_labelsr r� r � axis_test� s "r� c @ sr e Zd Zddddgfddddgfd e�dddg�fgZee� \ZZe j j d e e d �e j j d eed �d d� ��ZdS )� TestPlotBytes� string listr! r# r? � bytes list� a� b� c� bytes ndarrayr� r �bdatac C s, t �dddg�}||||� t|j|� d S �Nr r� r� �r7 rj r� �xaxis)r r^ r� r� �countsr r r �test_plot_bytes� s z TestPlotBytes.test_plot_bytesN)r0 r1 r2 r7 rj Z bytes_casesr4 Z bytes_ids� bytes_datar+ r5 r6 r� r� r� r r r r r� � s � r� c @ s� e Zd Zddddgfde�dddg�fdddd gfd e�ddd g�fgZee� \ZZe j j d e e d �e j j d eed �dd� ��ZdS )�TestPlotNumliker� r@ Z11�3zstring ndarrayr� � 1s 11� 3r� r� r �ndatac C s, t �dddg�}||||� t|j|� d S r� r� )r r^ r� r� r� r r r �test_plot_numlike� s z!TestPlotNumlike.test_plot_numlikeN)r0 r1 r2 r7 rj Z numlike_casesr4 Z numlike_idsZ numlike_datar+ r5 r6 r� r� r� r r r r r� � s � r� c @ s� e Zd Zejjdeed�dd� �Zej dd� �Z ej� d�ejjdeed�dd � ��Z ej� d�ejjdeed�d d � ��Z ej� d�ejjdeed�d d � ��Zejjdeed�dd� �ZdddgfdddgfdddgfddejgfgZee� \ZZejejejejjd�ejejejjd�gZ d d d gZ ejjde e d�ejjd eed�d d!� ��Z ejjde e d�ejjd eed�d"d#� ��Z d$S )%� TestPlotTypesr� r c C s( ddg}|||ddg� t |j|� d S )Nr� r� r r )r� r� )r r^ r� Zwordsr r r �test_plot_unicode� s z TestPlotTypes.test_plot_unicodec C s4 dddg| _ dddg| _ddd g| _dd d g| _d S ) Nr� Zhappyr� r r� r �Python�isZfunr r� )�x�xy�y�yxrv r r r � test_data� s zTestPlotTypes.test_datar� c C s" ||| j | j� t|j| j � d S r )r� r� r� r� �r r^ r� r� r r r �test_plot_xaxis� s z TestPlotTypes.test_plot_xaxisc C s" ||| j | j� t|j| j� d S r )r� r� r� r� r� r r r �test_plot_yaxis� s z TestPlotTypes.test_plot_yaxisc C s0 ||| j | j� t|j| j � t|j| j� d S r )r� r� r� r� r� r� r r r �test_plot_xyaxis� s z TestPlotTypes.test_plot_xyaxisc C st ||ddgddg� ||dddgdddg� ||dddgdddg� t |jddddg� t |jddddddg� d S )Nr! r# r$ �gr" �fr? )r� r� r� r� r r r �test_update_plot� s z TestPlotTypes.test_update_plotr r r' znumber integerr@ r rZ r[ r\ ZmissingZ12)Zmarksr} r~ r �xdatac C s* t �t�� |||ddg� W 5 Q R X d S )Nr r �r+ r, r- �r r^ r� r� r r r �test_mixed_type_exception s z'TestPlotTypes.test_mixed_type_exceptionc C s> t �t��* ||ddgddg� |||ddg� W 5 Q R X d S )Nr r r r r� r� r r r � test_mixed_type_update_exception s z.TestPlotTypes.test_mixed_type_update_exceptionN) r0 r1 r2 r+ r5 r6 r� r� r� r{ r� Z usefixturesr� r� r� r� r7 r( r8 r4 r9 rz r r} Zparamr~ Zxfailr ZPLOT_BROKEN_LISTZPLOT_BROKEN_IDSr� r� r r r r r� � s@ � � r� ) rx r+ Znumpyr7 Zmatplotlib.axesr Zmatplotlib.pyplotZpyplotr| Zmatplotlib.category�categoryr rs Z_textr� �objectr r: r= r{ r^ r} r~ r r� r� r� r� r� r� r� r� r r r r �<module> s& .>