Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/formats/style/__pycache__/test_style.cpython-37.pyc
Назад
B �5�g�� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dl m Z e� d�Z d dlmZ d dlmZmZmZmZ ejdd� �Zejdd � �Zejd d � �Zej�d d d ddd�dddd�gfdd ddd�d ddd�gfg�dd� �Zej�dd d ddd�dddd�gfdd ddd�d ddd�gfg�dd� �Z dd� Z d d � Z d d � Z d d!� Z ej�d"d dg�ej�d#d dg�ej�d$d dg�d%d&� ���Z!G d'd(� d(�Z"dS ))� N)� DataFrame� MultiIndex�jinja2)�Styler)�_get_level_lengths�_get_trimming_maximums�maybe_convert_css_to_tuples�non_reducing_slicec C s< t ddgddggt�dgddgg�t�dgd d gg�td �S ) N� � � � �i0�i1_a�i1_b�c0�c1_a�c1_b)�index�columnsZdtype)r r � from_product�int� r r �^/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/formats/style/test_style.py�mi_df s r c C s t | dd�S )Nr )�uuid_len)r )r r r r � mi_styler$ s r c C s� d| _ d| _| �d� | �ddd�g� | �� | �dg� | �� | �dg� | �d � | jd d d � | jd d � | � t ddgddgg| j | j d�� | � t ddgddgg| j | j d�� | S )N� Zabcde_Zcapt�aza:v;)�selector�props)r r )r r z class="box"�MISSINGr )�na_rep� precision)�axis�b�c)r r Za2�b2�c2)r �uuid� set_caption�set_table_styles� hide_columns� hide_index�set_table_attributes�format� highlight_max�set_td_classesr r r � set_tooltips)r r r r �mi_styler_comp) s( r3 zsparse_columns, exp_colsTz colspan="2"r )� is_visible� attributes�valueF� c C s� dddd�}dddd�}| � d|�}|d �� |d d d �� ksHt�|d �� |d d d �� kslt�|�� |d d d �� ks�t�|�� |d d d �� ks�t�d S ) NTr7 r )r4 r5 � display_valuer r �headr r r )� _translate�items�AssertionError)r Zsparse_columnsZexp_colsZ exp_l1_c0Z exp_l1_c1�ctxr r r � test_mi_styler_sparsify_columnsF s $$ r> zsparse_index, exp_rowsz rowspan="2"r c C s� dddd�}dddd�}| � |d�}|d �� |d d d �� ksHt�|d �� |d d d �� kslt�|�� |d d d �� ks�t�|�� |d d d �� ks�t�d S ) NTr7 r )r4 r5 r8 r r �bodyr )r: r; r< )r Z sparse_indexZexp_rowsZ exp_l1_r0Z exp_l1_r1r= r r r � test_mi_styler_sparsify_indexe s $$ r@ c C s� t �dd�� | �� }W d Q R X t �dd�� | �� }W d Q R X ||ksLt�t �dd�� | �� }W d Q R X t �dd�� | �� }W d Q R X ||ks�t�d S )Nzstyler.sparse.indexFTzstyler.sparse.columns)�pd�option_context�renderr< )r Zhtml1Zhtml2r r r � test_mi_styler_sparsify_options� s rD c C sL t ddddd�\} }| |fdks$t�t ddddd�\} }| |fdksHt�d S ) N�d g �?)Zscaling_factor)� � i� r i� )� r )r r< )ZrnZcnr r r �test_trimming_maximum� s rI c C s t t�d��dd��} t�dd�� | j�dd�}W d Q R X t|d d �d ksTt �t|d �d ksht �t|d d �d ks�t �t t�d��d d ��} t�dd�� | j�dd�}W d Q R X t|d d �d ks�t �t|d �d ks�t �t|d d �d k�st �d S )N�x �< r zstyler.render.max_elementsrG Tr9 r r r? r rF � ) r �np�arange�reshaperA rB �styler: �lenr< )�dfr= r r r �test_render_trimming� s rS c C s6 t �ddgdddgg�} tt�d��dd�| | d�}t�dd�� |j� d d �}W d Q R X t |d d �d ksrt �d di� � |d d d � � ks�t �ddi� � |d d d � � ks�t �ddi� � |d d d � � ks�t �t |d �dks�t �t |d d �d k�s t �d di� � |d d d � � k�s2t �d S )Nr r r �$ rG )r r zstyler.render.max_elementsr Tr? r r r5 z rowspan="2"�classzdata row0 col_trimzdata row_trim col_trimr9 z colspan="2") r r r rM rN rO rA rB rP r: rQ r< r; )ZmidxrR r= r r r �test_render_trimming_mi� s $$$rV � comprehensiverC �deepcopyc sN | r|n|}d|_ |r t�|�nt�|�}||k s6t�|rB|�� ddddg� |s�x�� fdd�|jD �D ]$}tt||��tt||��ksht�qhW n�dd d d d d dddg �x,�D ]$}tt||��tt||��ks�t�q�W xr� �fdd�|jD �D ]X}t||�d k�r&tt||��tt||��k�sFt�q�tt||��tt||��ks�t�q�W d S )Nr r" r# r) Z cellstyle_mapc s g | ]}t |�s|� kr|�qS r )�callable)�.0r )�exclr r � <listcomp>� s z test_copy.<locals>.<listcomp>�datar r r �caption�cell_ids� hide_index_Z hide_columns_�table_attributesc s( g | ] }t |�s|� kr|�kr|�qS r )rY )rZ r )r[ �shallowr r r\ � s )r �copyrX r< Zto_html�__dict__�id�getattr)rW rC rX r r3 �styler�s2�attrr )r[ rb r � test_copy� s6 & $ $rj c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zej�d'ejd(d(� ejd(d(�d)gf ejd*gd(d(�f ejd*gd)gf ejd(d+�d)d,gf g�ej�d-d.d*g�d/d0� ��Zej�d'ejd(d(� ejd(d(�d)gf ejd*gd(d(�f ejd*gd)gf ejd(d+�d)d,gf g�d1d2� �Z ej�d'ejd(d(�ejd3 f ejd(d(�ejd(d(�d)f f ejd(d(�ejd(d(�d)d4gf f ejejd5 d(d(�f ejejd(d(�d*f d(d(�f ejejd(d(�d*d6gf d(d(�f ejd(d(�d3f ejd5d(d(�f g�d7d8� �Z d9d:� Z d;d<� Z ej�d'ejd(d(� ejd(d(�d)gf ejd*gd(d(�f ejd*gd)gf ejd(d+�d)d,gf g�d=d>� �Z d?d@� Z!dAdB� Z"dCdD� Z#dEdF� Z$dGdH� Z%dIdJ� Z&dKdL� Z'dMdN� Z(dOdP� Z)dQdR� Z*dSdT� Z+dUdV� Z,dWdX� Z-dYdZ� Z.d[d\� Z/d]d^� Z0d_d`� Z1dadb� Z2dcdd� Z3dedf� Z4dgdh� Z5didj� Z6dkdl� Z7dmdn� Z8dodp� Z9dqdr� Z:dsdt� Z;dudv� Z<dwdx� Z=dydz� Z>d{d|� Z?d}d~� Z@ej�deAd�d�geBjCd(ggd)d,gd�d�gd��eAd�ggd,gd�gd��eAd�d�ggd,d4gd�gd��g�d�d�� �ZDd�d�� ZEd�d�� ZFd�d�� ZGej�d�d*d�d�d�d�g�d�d�� �ZHej�d�d�d�d(g�d�d�� �ZIej�d�ejd(d(�d(d(�f ejd(d(�d*f ejd*d(d(�f ejd*gd*gf ejd*d*gf ejd*gd*f ejd* ejd� eJd(d(d(�d.d*geB�Kd.d*g�e�Ld.d*g�g �d�d�� �ZMej�d�eNejLeBjKg�d�d�� �ZOd�d�� ZPej�d'ejd(d(�d(d(�f ejd(d(�ejd�g f ejd(d(�ejd�gd�gf f ejd(d(�ejd�d�d(d(�f f ejd(d(�ejd�d(d(�d�f f ejd(d(�ejd(d(�d�d�f f ejd(d(�ejd�d�d�gd(d(�f f ejd(d(�ejd�d�d�d�gd(d(�f f ejd(d(�ejd�d�d�d�gd�f f ejejd�g d(d(�f ejejd�gd�gf d(d(�f ejejd�d�d(d(�f d(d(�f ejejd�d(d(�d�f d(d(�f ejejd(d(�d�d�f d(d(�f ejejd(d(�d�d�d�gf d(d(�f ejejd(d(�d�d�d�d�gf d(d(�f ejejd�d�d�d�d�gf d(d(�f ejejd(d(�d�d�f ejd�d�d(d(�f f g�d�d�� �ZQd(S )�� TestStylerc C s� t j�d� tdt j�td��i�| _tddgt j�d�d��| _dd � | _ d d � | _ dd d �}|| _ t | j�| _ tdddgi�| _| jtddgddgt�ddg�d��g| _d| _d S )N� �ArG r r r )rm �Bc S s | S )Nr )�xr r r �<lambda>� � z)TestStyler.setup_method.<locals>.<lambda>c S s | S )Nr )ro r r r rp � rq �barc S s t jd|� �| j| jd�S )Nzcolor: )r �name)rA �Seriesr rs )ro �foor r r �h� s z"TestStyler.setup_method.<locals>.hz color: redz color: blueg �?g @r r% )�f�or& z )rr )rM �random�seedr Z permutation�range�sZrandnrR rw �grv r rg �attrsrA Z CategoricalZ dataframes� blank_value)�self�methodrv r r r � setup_method� s $zTestStyler.setup_methodc C s0 d}t jt|d�� tdddg� W d Q R X d S )Nz&``data`` must be a Series or DataFrame)�matchr r r )�pytest�raises� TypeErrorr )r� �msgr r r �test_init_non_pandas� s z TestStyler.test_init_non_pandasc C s& t t�ddg��}|jjdks"t�d S )Nr r )r rA rt r] �ndimr< )r� �resultr r r �test_init_series s zTestStyler.test_init_seriesc C s | j �� d S )N)rg � _repr_html_)r� r r r �test_repr_html_ok s z TestStyler.test_repr_html_okc C s@ d| j �� kst�t�dd�� d| j �� ks2t�W d Q R X d S )NZtex2jax_ignorezdisplay.html.use_mathjaxF)rg r� r< rA rB )r� r r r �test_repr_html_mathjax s z!TestStyler.test_repr_html_mathjaxc C s0 | j �| j� dgdgd�}| j j|ks,t�d S )N)�color�red)r� �blue))r r )r r )rg � _update_ctxr~ r= r< )r� �expectedr r r �test_update_ctx s zTestStyler.test_update_ctxc C sB t dddgi�}| j�|� ddgddgd�}| jj|ks>t�d S ) Nrm zcolor: red; foo: barzcolor:blue ; foo: baz;)r� r� )ru rr )r� r� )ru �baz))r r )r r )r rg r� r= r< )r� r~ r� r r r �/test_update_ctx_flatten_multi_and_trailing_semi s z:TestStyler.test_update_ctx_flatten_multi_and_trailing_semic C s2 t dd dgi�}t dd dgi�}| jj�� �|��|�}|�� �d�}t|j �dksXt �|j sbt �t|j �dkstt �|j dks�t �t|j�dks�t �|�� }t|j�dks�t �t|j �dks�t �|�� t|j�dks�t �t|j �dks�t �|j r�t �t|j �dk�s t �|j dk�st �t|j�dk�s.t �d S )Nrm �ttzcls-ar TF)r rR rP r0 r2 r1 r- r, rQ �_todor< ZtooltipsZ cell_contextr` Zhidden_columns�_computer= �clear)r� r� �cssr| r r r � test_clear s& zTestStyler.test_clearc C s6 t dddgi�}dd� }t|dd��|�}|�� d S )Nrm r r c S s t jddg| jd�S )Nz color: redz color: blue)rs )rA rt rs )ro r r r rp ; rq z(TestStyler.test_render.<locals>.<lambda>�AB)r) )r r �applyrC )r� rR rP r| r r r � test_render9 s zTestStyler.test_renderc C s: t | jdd�jdd� dgd�}|�� d|�� ks6t�d S )Nr )r c S s dS )Nz color: red;r )ro r r r rp B rq z1TestStyler.test_multiple_render.<locals>.<lambda>rm )�subsetzO<style type="text/css"> #T__row0_col0, #T__row1_col0 { color: red; } </style>)r rR �applymaprC r< )r� r| r r r �test_multiple_render@ s z TestStyler.test_multiple_renderc C s> t � }t|�}|�� t dgd�j�� t dgd�j�� d S )Nr )r )r )r r rC rP )r� Zempty_df�esr r r �test_render_empty_dfsI s z TestStyler.test_render_empty_dfsc C s6 t dddgi�}dd� }t|dd��|�}|�� d S )Nrm r r c S s t jddg| jd�S )Nzcolor: red; border: 1pxzcolor: blue; border: 2px)rs )rA rt rs )ro r r r rp U s z/TestStyler.test_render_double.<locals>.<lambda>r� )r) )r r r� rC )r� rR rP r| r r r �test_render_doubleS s z TestStyler.test_render_doublec C s� t dddgi�}|jjddd��� j}ddg}||d �}|�� |�� ksLt�x2t|�� |�� �D ] \}}t |�t |�ks`t�q`W d S ) Nrm r r �white�10px)r� �size)r� r� )r� r� ))r r )r r ) r rP �set_propertiesr� r= �keysr< �zip�values�sorted)r� rR r� �vr� Zv1Zv2r r r �test_set_properties\ s z TestStyler.test_set_propertiesc C sF t dddgi�}|jjtjd dd��� j}ddgi}||ksBt�d S ) Nrm r r )r rm r� )r� r� )r r )r� r� )r rP r� rA � IndexSlicer� r= r< )r� rR r� r� r r r �test_set_properties_subsetf s z%TestStyler.test_set_properties_subsetc C s� t ddgddgddgd��}|j�dd�}d d | jd| jd �d d d d ddd�ddd dddd�ddd dddd�gg}|d |ks�t�d S )Nr r r r r rG )rm rn �CTz blank level0�th)rU �typer6 r4 r8 zcol_heading level0 col0rm r7 )rU r8 r� r6 r4 r5 zcol_heading level0 col1rn zcol_heading level0 col2r� r9 )r rP r: r r< )r� rR r� r� r r r �$test_empty_index_name_doesnt_displayp s4 z/TestStyler.test_empty_index_name_doesnt_displayc C s� t ddgddgddgd��}|�d�j�d d �}d d | j| jd d �d d ddd dd�dd ddd dd�gdd dd dd�dd | jd | jd�dd | jd | jd�gg}|d |ks�t�d S )Nr r r r r rG )rm rn r� rm Tz blank level0r� )rU r� r6 r8 r4 zcol_heading level0 col0rn r7 )rU r� r6 r8 r4 r5 zcol_heading level0 col1r� zindex_name level0)rU r� r6 r4 r8 z blank col0z blank col1r9 )r � set_indexrP r: r r< )r� rR r� r� r r r �test_index_name� sF zTestStyler.test_index_namec C s� t ddgddgddgd��}|�dd g�j�d d �}d d | j| jd d �dd | j| jd d �dd ddd dd�gdd dd dd�dd d d d d�dd | jd | jd�gg}|d |ks�t�d S )Nr r r r r rG )rm rn r� rm rn T�blankr� )rU r� r6 r8 r4 z blank level0zcol_heading level0 col0r� r7 )rU r� r6 r8 r4 r5 zindex_name level0)rU r� r6 r4 r8 zindex_name level1z blank col0r9 )r r� rP r: r r< )r� rR r� r� r r r �test_multiindex_name� sD z TestStyler.test_multiindex_namec C s$ t ddddgi�}|j�dd� d S )Nr r r r T)r rP r: )r� rR r r r �test_numeric_columns s z TestStyler.test_numeric_columnsc C s� t ddgddgd��}dd� }|jj|dd�}t|j�dks@t�t|j�dksRt�|�� dgdgdgdgd�}|j|ks~t�|jj|dd�}d gdgd gdgd�}|�� |j|ks�t�|j�|�}|�� |j|ks�t�d S ) Nr r )rm rn c s � fdd�� D �S )Nc s g | ]}d � � � � ��qS )zval: )�max)rZ r� )ro r r r\ s z@TestStyler.test_apply_axis.<locals>.<lambda>.<locals>.<listcomp>r )ro r )ro r rp rq z,TestStyler.test_apply_axis.<locals>.<lambda>)r$ )�val�1))r r )r r )r r )r r )r� �0)r rP r� rQ r� r< r= r� )r� rR rw r� r� r r r �test_apply_axis s* zTestStyler.test_apply_axis�slice_Nrm r r rn r$ r c sJ � j jj� j|�dd��� j}� �fdd�t� j j�D �}||ksFt�d S )Nr� )r$ r� ru c sV i | ]N\}}t � jj�D ]8\}}|� jj� jkr|� jj� jkrd g||f�qqS ))r� r� )� enumeraterR r �locr )rZ �r�rowr& �col)r� r� r r � <dictcomp>C s z0TestStyler.test_apply_subset.<locals>.<dictcomp>) rR rP r� rv r� r= r� r r< )r� r� r$ r� r� r )r� r� r �test_apply_subset2 s z TestStyler.test_apply_subsetc sH � j jjdd� �d��� j}� �fdd�t� j j�D �}||ksDt�d S )Nc S s dS )Nz color:baz;r )ro r r r rp W rq z1TestStyler.test_applymap_subset.<locals>.<lambda>)r� c sV i | ]N\}}t � jj�D ]8\}}|� jj� jkr|� jj� jkrd g||f�qqS ))r� r� )r� rR r r� r )rZ r� r� r& r� )r� r� r r r� Y s z3TestStyler.test_applymap_subset.<locals>.<dictcomp>)rR rP r� r� r= r� r r< )r� r� r� r� r )r� r� r �test_applymap_subsetK s z TestStyler.test_applymap_subset)ro rm r� )r r r c C s` t �ddgddgg�}t �ddgddgg�}ttj�d d �||d �}|jjd d � |d ��� d S )Nr r% r r ro �yrm rn r )r r c S s dS )Nz color: red;r )ro r r r rp t rq z<TestStyler.test_applymap_subset_multiindex.<locals>.<lambda>)r� ) r r r rM ry ZrandrP r� rC )r� r� �idxr� rR r r r � test_applymap_subset_multiindexa s z*TestStyler.test_applymap_subset_multiindexc C s� t �ddddgddddgg�}tddgddgg|ddgd�}tdd ddgd dddggd d g|d �}tjd d �tjd d �dd�f f }d d� }|j| |jj||d� d S )Nr r r r% �%�#r7 )Zlevels�codes�names���ZhelloZworld)r r c S s | dk r dnd}d|� �S )Nr r� �blackzcolor: r )r� r� r r r �color_negative_red� s zKTestStyler.test_applymap_subset_multiindex_code.<locals>.color_negative_red)r� ) rM �arrayr r rA r� r� rP r� )r� r� r rR Z pct_subsetr� r r r �$test_applymap_subset_multiindex_codev s $$ z/TestStyler.test_applymap_subset_multiindex_codec sd dd� � d}t �t�� �jj�� |��� j}W d Q R X � �fdd�t�jj �D �}||ks`t �d S )Nc S s | dkS )Ng �?r )ro r r r rw � s z/TestStyler.test_where_with_one_style.<locals>.fzfoo: barc sF i | ]>\}}t �jj�D ](\}}� �jj||f �rd g||f�qqS ))ru rr )r� rR r r� )rZ r� r� r& r� )rw r� r r r� � s z8TestStyler.test_where_with_one_style.<locals>.<dictcomp>) �tm�assert_produces_warning� FutureWarningrR rP �wherer� r= r� r r< )r� �style1r� r� r )rw r� r �test_where_with_one_style� s z$TestStyler.test_where_with_one_stylec sp dd� � d}d}t �t��" �jjj� ||�d��� j}W d Q R X � ��fdd�t�jj �D �}||kslt �d S )Nc S s | dkS )Ng �?r )ro r r r rw � s z'TestStyler.test_where_subset.<locals>.fzfoo: barzbaz: foo)r� c sn i | ]f\}}t �jj�D ]P\}}|�jj� jkr|�jj� jkr� �jj||f �r\d ndg||f�qqS ))ru rr )r� ru )r� rR r r� r )rZ r� r� r& r� )rw r� r� r r r� � s z0TestStyler.test_where_subset.<locals>.<dictcomp>) r� r� r� rR rP r� r� r= r� r r< )r� r� r� �style2�resr� r )rw r� r� r �test_where_subset� s &z TestStyler.test_where_subsetc s� dd� � d�d�� ��fdd�}t jd d � t jd d �dgf t jdgd d �f t jdgdgf t jd d �dd gf g}xb|D ]Z}t�t��" | jjj� ��|d ��� j }W d Q R X | jjj ||d ��� j }||ks~t �q~W d S ) Nc S s | dkS )Ng �?r )ro r r r rw � s z=TestStyler.test_where_subset_compare_with_applymap.<locals>.fzfoo: barzbaz: fooc s � | �r �S �S )Nr )ro )rw r� r� r r r} � s z=TestStyler.test_where_subset_compare_with_applymap.<locals>.grm r r rn )r� ) rA r� r� r� r� rR rP r� r� r= r� r< )r� r} Zslicesr� r� r� r )rw r� r� r �'test_where_subset_compare_with_applymap� s &z2TestStyler.test_where_subset_compare_with_applymapc C sr t ddgddgg�}dd� }t�t�� |jj|dddd ��� j}W d Q R X d gd gd gd gd �}||ksnt�d S ) Nr r r r c S s | |kS )Nr )ro r� r r r rw � s z'TestStyler.test_where_kwargs.<locals>.fz color:green;z color:red;)r� )r� r� )r� �green))r r )r r )r r )r r ) r r� r� r� rP r� r� r= r< )r� rR rw r� r� r r r �test_where_kwargs� s $ z TestStyler.test_where_kwargsc C sb t dddgi�}|j}dgdgd�|_|�dd�d }dgd gd �dgd gd �g}||ks^t�d S ) Nrm r r )r� r� )r7 r7 ))r r )r r T� cellstyle� row0_col0)r � selectors� row1_col0)r rP r= r: r< )r� rR r| r� r� r r r � test_empty� s zTestStyler.test_emptyc C sX t dddgi�}|j}dgdgd�|_|�dd�d }dgdd gd �g}||ksTt�d S ) Nrm r r )r� r� ))r r )r r Tr� r� r� )r r� )r rP r= r: r< )r� rR r| r� r� r r r �test_duplicate� s zTestStyler.test_duplicatec C sl t d d gddggddgd�}t|dd��dd�}|d d d d dksLt�|d d d d dksht�d S )Ng�������?g333333�?rm rn )r �NA)r" Tr? r r r8 r )r r r: r< )r� rR r= r r r �test_init_with_na_rep� s z TestStyler.test_init_with_na_repc C s� t d d gddggddgd�}t�t�� |j�d��dd�}W d Q R X |d d d d dksbt�|d d d d dks~t�t�t��( |j�d�jd d dgd��dd�}W d Q R X |d d d d dks�t�|d d d d d ks�t�d S )Ng�������?g333333�?rm rn )r r� Tr? r r r8 r �-)r" r� ) r r� r� r� rP Z set_na_repr: r< r/ )r� rR r= r r r �test_set_na_rep s zTestStyler.test_set_na_repc C s^ t | jdd�}|�� }td|kd|kg�s.t�| jj}|�d�}||ksLt�|jdksZt�d S )Nru )r^ r^ r� )r rR rC �allr< rP r* r^ )r� rg r� r r r � test_caption s zTestStyler.test_captionc C sR t | jdd�}|�� }d|ks"t�| jj}|�d�}||ks@t�|jdksNt�d S )NZabc123)r) Zaaa)r rR rC r< rP Zset_uuidr) )r� rg r� r r r � test_uuid s zTestStyler.test_uuidc C s` t ddddgddddgd ��}|jjd d �}d |ks8t�t�d |�}t�|�jt |�ks\t�d S ) Nr r r rG r r rF � )r r% �test)r) z id="(.*?)") r rP rC r< �re�findallrM �uniquer� rQ )r� rR r� Zidsr r r �test_unique_id& s zTestStyler.test_unique_idc C s� ddgd�g}t | j|d�}d�|�� �� �}d|ks:t�| jj}|�|�}||ksXt�|j|ksft�ddd�g}| jj�|�}d�|�� �� �}d|ks�t�d S )Nr� )ru rr )r r )� table_styles� zth { foo: bar; }zfoo:bar;) r rR �joinrC �splitr< rP r+ r� )r� rP rg r� r r r �test_table_styles. s z TestStyler.test_table_stylesc C sX | j j�ddd�ddd�g��dd�d }|dd gd�d d gd�dd gd�gksTt�d S ) Nzth,tdz color:red;)r r �trz color:green;Tr� r� )r� r� �td)r� r� )rR rP r+ r: r< )r� r= r r r �test_table_styles_multiple? s z%TestStyler.test_table_styles_multiplec C s@ ddg}t d�|kst�t d�|ks(t�g }t d�|ks<t�d S )N)r r% )r& zd ez a:b;c:d e;za: b ;c: d e r7 )r r< )r� r� r r r � test_maybe_convert_css_to_tuplesL s z+TestStyler.test_maybe_convert_css_to_tuplesc C s* d}t jt|d�� td� W d Q R X d S )Nz6Styles supplied as string must follow CSS rule formats)r� �err)r� r� � ValueErrorr )r� r� r r r �$test_maybe_convert_css_to_tuples_errS s z/TestStyler.test_maybe_convert_css_to_tuples_errc C sH d}t | j|d�}|�� }d|ks&t�| jj�|��� }d|ksDt�d S )Nzclass="foo" data-bar)ra )r rR rC r< rP r. )r� r5 rg r� r r r �test_table_attributesX s z TestStyler.test_table_attributesc C sZ t | jdd�}|jdks t�t�t�� |�d�}W d Q R X ||ksHt�|jdksVt�d S )Nr )r# r )r rR r# r< r� r� r� Z set_precision)r� r| rh r r r �test_precisiona s zTestStyler.test_precisionc C sD dd� }t ddgddgg�jj|d d��� j}|d d gks@t�d S ) Nc S s$ t t�| | �� kdd�| j| jd�S )Nz color: redr7 )r r )r rM r� r� r r )ro r r r rw k s z%TestStyler.test_apply_none.<locals>.fr r r r )r$ )r r )r� r� )r rP r� r� r= r< )r� rw r� r r r �test_apply_nonej s &zTestStyler.test_apply_nonec C sL | j j�� }|�d�dks t�| j j�� �� }|�d�t| j j�ksHt�d S )Nr� r )rR rP rC �countr< r0 rQ r )r� r� r r r � test_trimu s zTestStyler.test_trimc C sh dd� }dd� }| j }|�|�j|dd��� �� |�� }| jj}|�|� |j|jks\t �|� � d S )Nc S s | dkr dS dS )Nr z color: redz color: bluer )ro r r r rp } rq z(TestStyler.test_export.<locals>.<lambda>c S s | dkrd|� �S d|� �S )Nr zcolor: r )ro �zr r r rp ~ rq r% )r ) rg r� r0 r� ZexportrR rP Zuser� r< rC )r� rw r} r� r� r� r r r � test_export| s zTestStyler.test_exportc C sX t ddgddgg�}d}tjt|d��, |jjdd� tjd dgd d �f d � W d Q R X tjt|d��, |jjd d� tjd dgd d �f d � W d Q R X tjt|d�� |j�d d� � W d Q R X tjt|d�� |jjd d� dd � W d Q R X d}tjt|d�� |jjdd� dd� W d Q R X d}tjt|d�� |jjdd� d d� W d Q R X d S )Nr r r r zreturned the wrong shape)r� c S s dS )Nro r )ro r r r rp � rq z1TestStyler.test_bad_apply_shape.<locals>.<lambda>r )r� c S s dgS )Nr7 r )ro r r r rp � rq c S s ddddgS )Nr7 r )ro r r r rp � rq c S s dddgS )Nr7 r )ro r r r rp � rq z-Length mismatch: Expected axis has 3 elementsc S s dddgS )Nr7 r )ro r r r rp � rq )r$ z!returned ndarray with wrong shapec S s t �dgdgg�S )Nr7 )rM r� )ro r r r rp � rq )r r� r� r� rP �_applyrA r� )r� rR r� r r r �test_bad_apply_shape� s 00 z TestStyler.test_bad_apply_shapec C sN dd� }t ddgddgg�}d}tjt|d�� |jj|d d � W d Q R X d S ) Nc S s dS )Nr7 r )ro r r r rw � s z+TestStyler.test_apply_bad_return.<locals>.fr r r r zOmust return a DataFrame or ndarray when passed to `Styler.apply` with axis=None)r� )r$ )r r� r� r� rP r )r� rw rR r� r r r �test_apply_bad_return� s z TestStyler.test_apply_bad_returnc C sN dd� }t ddgddgg�}d}tjt|d�� |jj|d d � W d Q R X d S ) Nc S s t ddgddgd�S )Nr r r r% )r r )r )ro r r r rw � s z+TestStyler.test_apply_bad_labels.<locals>.fr r r r z2must have identical index and columns as the input)r� )r$ )r r� r� r� rP r )r� rw rR r� r r r �test_apply_bad_labels� s z TestStyler.test_apply_bad_labelsc C s� t �ddgdddgg�}ddddddddd�}t|dd d �}t�||� ddddddddddddd � }t|d d d �}t�||� d S ) Nr r% r r r r ))r r )r r )r r )r r )r r )r r )r r )r r TrE )�sparsify� max_index) )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r )r r F)r r r r� �assert_dict_equal)r� r r� r� r r r �test_get_level_lengths� s2 z!TestStyler.test_get_level_lengthsc C s� t �ddddgddddgg�}dddddddd�}t|ddd �}t�||� ddddddddd �}t|d dd �}t�||� d S ) Nr r r r% �d))r r )r r )r r )r r )r r )r r )r r TrE )r r ))r r )r r )r r )r r )r r )r r )r r )r r F)r � from_arraysr r� r )r� r r� r� r r r � test_get_level_lengths_un_sorted� s( z+TestStyler.test_get_level_lengths_un_sortedc C s� t dddgitjddgddggddgd�d �}|j�d d �}|d d }d dd d dd�ddd d dd�d| jd d | jd�g}||ks�t�d S )Nrm r r r r � idx_level_0� idx_level_1)r� )r Tr9 zindex_name level0r� )rU r6 r� r4 r8 zindex_name level1z blank col0)r r r rP r: r r< )r� rR r� r9 r� r r r �test_mi_sparse_index_names� s, z%TestStyler.test_mi_sparse_index_namesc C s� t t�d��dd�tjddddgddddggdd gd �tjd d d d gddddggd dgd �d�}|j�dd�}|d d }d| j| jddd�dddddd�ddddddd�ddddddd�ddddddd�d dddddd�g}||ks�t �d S ) N� r r r% r r r r r )r� ZC1ZC2Zcol_0Zcol_1)r r Tr9 r� r� )rU r6 r8 r� r4 zindex_name level1)rU r6 r8 r4 r� zcol_heading level1 col0r7 )rU r8 r4 r� r6 r5 zcol_heading level1 col1zcol_heading level1 col2zcol_heading level1 col3) r rM rN rO r r rP r: r r< )r� rR r� r9 r� r r r �test_mi_sparse_column_names sX & z&TestStyler.test_mi_sparse_column_namesc C s\ | j �� �dd�}t|d �dks&t�d| jj_| jj�� �dd�}t|d �dksXt�d S )NTr9 r Z some_name) rg r, r: rQ r< rR r rs rP )r� r= r r r �test_hide_column_headersR s z#TestStyler.test_hide_column_headersc C sF | j j�dd�}|d d d d s(t�|d d d d s@t�| j j�� �dd�}|d d d d rlt�|d d d d r�t�| j �d�j�dd�}|d d d d s�t�t|d �dks�t�|d d d d s�t�| j �d�j�� �dd�}|d d d d �rt�t|d �dk�s(t�|d d d d �rBt�d S ) NTr? r r4 r9 rm r r )rR rP r: r< r- r� rQ )r� r= �ctx2Zctx3Zctx4r r r �test_hide_single_indexZ s z!TestStyler.test_hide_single_indexc C s t dddgitjddgddggddgd�d �}|j�d d �}|d d d d sTt�|d d d d slt�|d d d d s�t�|d d d d s�t�|j�� �d d �}|d d d d r�t�|d d d d r�t�|d d d d r�t�|d d d d �rt�d S )Nrm r r r r r r )r� )r Tr? r4 r9 )r r r rP r: r< r- )r� rR Zctx1r r r r �test_hide_multiindexo s z TestStyler.test_hide_multiindexc C s� | j j�dd�}|d d d d s(t�|d d d d dksDt�|d d d d s\t�|d d d d d ksxt�|d d d d s�t�|d d d d s�t�| j j�d��dd�}|d d d d r�t�|d d d d r�t�|d d d d �st�| j j�dd g��dd�}|d d d d �r<t�|d d d d �rVt�|d d d d �rpt�|d d d d �r�t�d S ) NTr9 r r r4 r8 rm r rn r? )rR rP r: r< r, )r� r= r r r � test_hide_columns_single_level� s z)TestStyler.test_hide_columns_single_levelc C s� t jddgddggddgd�}t jddgddggdd gd�}tdd gd d gg||d �}|j�dd�}|d d d d s|t�|d d d d s�t�|d d d d dks�t�|d d d d s�t�|d d d d s�t�|d d d d d ks�t�|d d d d �st�|d d d d d k�s4t�|j�d��dd�}|d d d d �rbt�|d d d d �r|t�|d d d d �r�t�|d d d d �s�t�|j�dg��dd�}|d d d d �r�t�|d d d d �s�t�|d d d d �rt�|d d d d �r.t�|d d d d �sHt�|d d d d d k�sft�|j�dg��� �dd�}|d d d d �r�t�|d d d d �s�t�|d d d d �s�t�|d d d d �r�t�|d d d d �rt�|d d d d �s t�|d d d d d k�s:t�|j�d��dd�}xBdD ]:}|d d | d �rrt�|d d | d �rTt��qTW |j�d��dd�}xBdD ]:}|d d | d �r�t�|d d | d �s�t��q�W d S )Nr r r r r )r� r% Z col_level_0Z col_level_1r r r )r r Tr9 r4 r8 r? )r% r )r% r )r r r r )r r )r r r rP r: r< r, r- )r� �i1�i2rR r= �ir r r �#test_hide_columns_index_mult_levels� sT z.TestStyler.test_hide_columns_index_mult_levelsc C sf dd� }| j jj|ddd�}d|�� ks,t�dd� }| j j}|j|d fd d d �}|d d |fksbt�d S ) Nc S s | � d|� d|� ��S )NzDataframe with a = z and b = )r* )rg r r% r r r �set_caption_from_template� s z7TestStyler.test_pipe.<locals>.set_caption_from_templaterm rn )r% z Dataframe with a = A and b = Bc S s | ||fS )Nr )r r% rg r r r rw � s z TestStyler.test_pipe.<locals>.frg r r )r r% )rR rP �piperC r< )r� r rg rw r� r r r � test_pipe� s zTestStyler.test_pipec C sB t dggd�}t|ddd�}|�� |�� }|�d�dks>t�d S )Nr )r] �_F)r) r_ z <td class="data row0 col0" >r� )r r rC �findr< )r� rR rg r| r r r �test_no_cell_ids� s zTestStyler.test_no_cell_ids�classesr7 z test-classr r% )r] r r Zunusedc C s� t ddgddggddgddgd �}t|dd d ��|��� }d |ksFt�d |ksRt�d|ks^t�d|ksjt�t|ddd ��|��� }d|ks�t�d|ks�t�d|ks�t�d|ks�t�d S )Nr r r r rm rn r r% )r] r r F)r r_ z#<td class="data row0 col0" >0</td>z.<td class="data row0 col1 test-class" >1</td>z#<td class="data row1 col0" >2</td>z#<td class="data row1 col1" >3</td>Tz4<td id="T__row0_col0" class="data row0 col0" >0</td>z?<td id="T__row0_col1" class="data row0 col1 test-class" >1</td>z4<td id="T__row1_col0" class="data row1 col0" >2</td>z4<td id="T__row1_col1" class="data row1 col1" >3</td>)r r r1 rC r< )r� r& rR r| r r r �test_set_data_classes� s " z TestStyler.test_set_data_classesc C s� t dddgdddgddd ggdddgdddgd �}t d d gd dggddgddgd �}t|dd��|��� }d|ksvt�d|ks�t�d|ks�t�d|ks�t�d|ks�t�d S )Nr r r r r r rG � � )r] r r �mi�ma�mu�mo)r z7<td id="T__row0_col0" class="data row0 col0 mi" >0</td>z7<td id="T__row0_col2" class="data row0 col2 ma" >2</td>z4<td id="T__row1_col1" class="data row1 col1" >4</td>z7<td id="T__row2_col0" class="data row2 col0 mu" >6</td>z7<td id="T__row2_col2" class="data row2 col2 mo" >8</td>)r r r1 rC r< )r� rR r& r| r r r � test_set_data_classes_reindex s 0 z(TestStyler.test_set_data_classes_reindexc C s\ t ddgddggddgd�}|j�ddgd �g�jd d gd �gd d �}t|j�dksXt�d S )Nr r r rm rn )r] r r7 )zbackground-color�yellow)r r z.col0)zbackground-colorr� F)� overwrite)r rP r+ rQ r� r< )r� rR rg r r r �test_chaining_table_styles s z%TestStyler.test_chaining_table_stylesc C s� t ddgddggddgd�}t|dd�}|�ddd gd �gi�}d |�� ksPt�|jddd gd �gidd �}d |�� ks|t�d S )Nr r r rm rn )r] r )r r7 )r� r� )r r z #T__ .col0 { color: blue; })r$ z #T__ .row0 { color: blue; })r r r+ rC r< )r� rR r| r r r �test_column_and_row_styling* s z&TestStyler.test_column_and_row_styling�len_r � �! rE c C sj t dggd�}t||dd��� }|�d�}||d d � �d�}|dkrV|d ksft�n||d ksft�d S ) Nrm )r] F)r r_ zid="T_rG �"r4 r5 r )r r rC r$ r< )r� r3 rR r| Zstrt�endr r r � test_uuid_len5 s zTestStyler.test_uuid_len����badc C sB t dggd�}d}tjt|d�� t||dd��� W d Q R X d S )Nrm )r] z3``uuid_len`` must be an integer in range \[0, 32\].)r� F)r r_ )r r� r� r� r rC )r� r3 rR r� r r r �test_uuid_len_raisesA s z TestStyler.test_uuid_len_raises�slc)r r c C s4 t ddgddgg�}t|�}t|j| t �s0t�d S )Nr r r r )r r � isinstancer� r< )r� r< rR �tslice_r r r �test_non_reducing_sliceI s z"TestStyler.test_non_reducing_slice�boxc C s` |dg�}t ddgddgd�ddgd�}tjd d �dgf }t|�}t�|j| |j| � d S ) Nrm r r r r )rm rn rn )r )r rA r� r r� �assert_frame_equalr� )r� r@ r� rR r� r� r r r �test_list_slice` s zTestStyler.test_list_slicec C sz ddgddgddgddgd�}t |ddgd�}tj}|d d �|d f }t|�}|j| }t dddgi�}t�||� d S ) Nr r r r ))r r )r r& )r% r& )r% r r )r )r% r )r rA r� r r� r� rA )r� ZdicrR r� r� r> r� r� r r r �%test_non_reducing_slice_on_multiindexk s z0TestStyler.test_non_reducing_slice_on_multiindexr& �er r� �U�W�Y�Zc C s| t �ddgddgddgg�}t �ddgd d gd d gg�}tt�d ��dd�||d�}|j| }|jt|� }t� ||� d S )Nr r% r& r rD rw rE �VrF �XrG rH �@ r) )r r ) r r r rM rN rO r� r r� rA )r� r� �colsZidxsrR r� r� r r r �+test_non_reducing_multi_slice_on_multiindex| s z6TestStyler.test_non_reducing_multi_slice_on_multiindex)R�__name__� __module__� __qualname__r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �mark� parametrizerA r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r r r r r r r" r% r rM �nanr' r. r1 r2 r8 r; �slicer� rt r? �listrB rC rM r r r r rk � s +;: " " " " $@= & $&" $&"rk )#rc r� �numpyrM r� ZpandasrA r r Zpandas._testingZ_testingr� Z importorskipr Zpandas.io.formats.styler Z pandas.io.formats.style_renderr r r r Zfixturer r r3 rQ rR r> r@ rD rI rS rV rj rk r r r r �<module> sH )
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка