Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/formats/__pycache__/test_to_latex.cpython-37.pyc
Назад
B �5�gE� � @ s$ d dl Z d dlmZ d dlmZ d dlZd dlZd dlmZmZ d dl m Z d dl m Z d dlmZmZmZmZ dd� Zejd d � �ZG d d � d �ZG d d� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�Z G dd� d�Z G dd � d �Z G d d � d �Z G d d � d �Z dS )!� N)�datetime)�dedent)� DataFrame�Series)�DataFrameFormatter)�RegularTableBuilder�RowBodyIterator�RowHeaderIterator�RowStringConverterc C s t | ��� S )z�Dedent without new line in the beginning. Built-in textwrap.dedent would keep new line character in the beginning of multi-line string starting from the new line. This version drops the leading new line character. )r �lstrip)�string� r �[/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/formats/test_to_latex.py�_dedent s r c C s t ddgddgd��S )z>Short dataframe for testing table/tabular/longtable LaTeX env.� � �b1�b2)�a�b)r r r r r �df_short! s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zej � d d d ddgddddhddd�g�dd� �Z dd� Z dd� Z dd� Zdd � Zd d � Zd d � Zd!S )"� TestToLatexc C sL t �d��8}|�|� t|��}|�� |�� ks4t�W d Q R X W d Q R X d S )Nztest.tex)�tm� ensure_clean�to_latex�open�read�AssertionError)�self� float_frame�path�fr r r �test_to_latex_to_file( s z!TestToLatex.test_to_latex_to_filec C sd t dgg�}t�d��D}|j|dd� tj|ddd��}|�� |�� ksLt�W d Q R X W d Q R X d S )Nu außgangenztest.texzutf-8)�encoding�r)r r r r �codecsr r r )r �dfr r! r r r �(test_to_latex_to_file_utf8_with_encoding. s z4TestToLatex.test_to_latex_to_file_utf8_with_encodingc C s` t dgg�}t�d��@}|�|� tj|ddd��}|�� |�� ksHt�W d Q R X W d Q R X d S )Nu außgangenztest.texr$ zutf-8)r# )r r r r r% r r r )r r& r r! r r r �+test_to_latex_to_file_utf8_without_encoding6 s z7TestToLatex.test_to_latex_to_file_utf8_without_encodingc C s6 t ddgddgd��}|�� }td�}||ks2t�d S )Nr r r r )r r z� \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& �result�expectedr r r � test_to_latex_tabular_with_index> s z,TestToLatex.test_to_latex_tabular_with_indexc C s: t ddgddgd��}|jdd�}td�}||ks6t�d S ) Nr r r r )r r F)�indexz� \begin{tabular}{rl} \toprule a & b \\ \midrule 1 & b1 \\ 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �#test_to_latex_tabular_without_indexO s z/TestToLatex.test_to_latex_tabular_without_index�bad_column_format� g333333�?�lr$ )r$ �cr1 )r r c C sD t ddgddgd��}d}tjt|d�� |j|d� W d Q R X d S ) Nr r r r )r r z$column_format must be str or unicode)�match)� column_format)r �pytest�raises� ValueErrorr )r r. r& �msgr r r � test_to_latex_bad_column_format` s z+TestToLatex.test_to_latex_bad_column_formatc C s |j dd� d S )N�lcr)r3 )r )r r r r r �&test_to_latex_column_format_just_worksj s z2TestToLatex.test_to_latex_column_format_just_worksc C s: t ddgddgd��}|jdd�}td�}||ks6t�d S ) Nr r r r )r r r9 )r3 z� \begin{tabular}{lcr} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �test_to_latex_column_formatn s z'TestToLatex.test_to_latex_column_formatc C s2 t ddg�}|jdjd�}td�}||ks.t�d S )Ng @�@�testz{:,.0f})� float_formatz� \begin{tabular}{ll} \toprule {} & 0 \\ \midrule 0 & 1,000 \\ 1 & test \\ \bottomrule \end{tabular} )r r �formatr r )r Zserr) r* r r r �%test_to_latex_float_format_object_col s z1TestToLatex.test_to_latex_float_format_object_colc C s& t � }|�� }td�}||ks"t�d S )Nz� \begin{tabular}{l} \toprule Empty DataFrame Columns: Index([], dtype='object') Index: Index([], dtype='object') \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �test_to_latex_empty_tabular� s z'TestToLatex.test_to_latex_empty_tabularc C s. t dddg�}|�� }td�}||ks*t�d S )Nr r r1 z� \begin{tabular}{ll} \toprule {} & 0 \\ \midrule 0 & a \\ 1 & b \\ 2 & c \\ \bottomrule \end{tabular} )r r r r )r �sr) r* r r r �test_to_latex_series� s z TestToLatex.test_to_latex_seriesc C s< t dddgi�}d|j_|jdd�}td�}||ks8t�d S )Nr r r ZfooF)Z index_namesz� \begin{tabular}{lr} \toprule {} & a \\ \midrule 0 & 1 \\ 1 & 2 \\ \bottomrule \end{tabular} )r r, �namer r r )r r& r) r* r r r � test_to_latex_midrule_location� s z*TestToLatex.test_to_latex_midrule_locationN)�__name__� __module__� __qualname__r" r' r( r+ r- r4 �mark� parametrizer8 r: r; r? r@ rB rD r r r r r ' s *r c @ s� e Zd Zdd� Zdd� Zdd� Zej�de dd d gi�d fe d d gd d gd ��d fe d d gd d gddgd��d fg�dd� �Z dS )�TestToLatexLongtablec C s* t � }|jdd�}td�}||ks&t�d S )NT)� longtablez� \begin{longtable}{l} \toprule Empty DataFrame Columns: Index([], dtype='object') Index: Index([], dtype='object') \\ \end{longtable} )r r r r )r r& r) r* r r r � test_to_latex_empty_longtable� s z2TestToLatexLongtable.test_to_latex_empty_longtablec C s: t ddgddgd��}|jdd�}td�}||ks6t�d S ) Nr r r r )r r T)rK a� \begin{longtable}{lrl} \toprule {} & a & b \\ \midrule \endfirsthead \toprule {} & a & b \\ \midrule \endhead \midrule \multicolumn{3}{r}{{Continued on next page}} \\ \midrule \endfoot \bottomrule \endlastfoot 0 & 1 & b1 \\ 1 & 2 & b2 \\ \end{longtable} )r r r r )r r& r) r* r r r �"test_to_latex_longtable_with_index� s z7TestToLatexLongtable.test_to_latex_longtable_with_indexc C s< t ddgddgd��}|jddd�}td �}||ks8t�d S ) Nr r r r )r r FT)r, rK a� \begin{longtable}{rl} \toprule a & b \\ \midrule \endfirsthead \toprule a & b \\ \midrule \endhead \midrule \multicolumn{2}{r}{{Continued on next page}} \\ \midrule \endfoot \bottomrule \endlastfoot 1 & b1 \\ 2 & b2 \\ \end{longtable} )r r r r )r r& r) r* r r r �%test_to_latex_longtable_without_index� s z:TestToLatexLongtable.test_to_latex_longtable_without_indexzdf, expected_numberr r r � � )r r r/ � )r r r1 c C s& |j ddd�}d|� d�|ks"t�d S )NFT)r, rK z \multicolumn{�})r r )r r& Zexpected_numberr) r r r �.test_to_latex_longtable_continued_on_next_page s zCTestToLatexLongtable.test_to_latex_longtable_continued_on_next_pageN) rE rF rG rL rM rN r4 rH rI r rS r r r r rJ � s rJ c @ sX e Zd Zdd� Zdd� Zdd� Zdd� Zej� d d gd fd d dg�dd� �Z dd� Z dS )�TestToLatexHeaderc C s: t ddgddgd��}|jdd�}td�}||ks6t�d S ) Nr r r r )r r F)�headerz� \begin{tabular}{lrl} \toprule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �"test_to_latex_no_header_with_index s z4TestToLatexHeader.test_to_latex_no_header_with_indexc C s< t ddgddgd��}|jddd�}td�}||ks8t�d S ) Nr r r r )r r F)r, rU z� \begin{tabular}{rl} \toprule 1 & b1 \\ 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �%test_to_latex_no_header_without_index/ s z7TestToLatexHeader.test_to_latex_no_header_without_indexc C s> t ddgddgd��}|jddgd�}td �}||ks:t�d S ) Nr r r r )r r �AA�BB)rU z� \begin{tabular}{lrl} \toprule {} & AA & BB \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �)test_to_latex_specified_header_with_index? s z;TestToLatexHeader.test_to_latex_specified_header_with_indexc C s@ t ddgddgd��}|jddgdd �}td �}||ks<t�d S ) Nr r r r )r r rX rY F)rU r, z� \begin{tabular}{rl} \toprule AA & BB \\ \midrule 1 & b1 \\ 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �,test_to_latex_specified_header_without_indexQ s z>TestToLatexHeader.test_to_latex_specified_header_without_indexzheader, num_aliases�Ar ))�Br ))�Col1�Col2�Col3rO ))r^ r_ r` ZCol4rP c C sL t ddgddgd��}d|� d�}tjt|d�� |j|d � W d Q R X d S ) Nr r r r )r r zWriting 2 cols but got z aliases)r2 )rU )r r4 r5 r6 r )r rU Z num_aliasesr& r7 r r r �8test_to_latex_number_of_items_in_header_missmatch_raisesc s zJTestToLatexHeader.test_to_latex_number_of_items_in_header_missmatch_raisesc C s: t ddgddgd��}|jdd�}td�}||ks6t�d S ) Ng �?g������ @r r )r r �,)�decimalz� \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1,0 & b1 \\ 1 & 2,1 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �test_to_latex_decimalw s z'TestToLatexHeader.test_to_latex_decimalN) rE rF rG rV rW rZ r[ r4 rH rI ra rd r r r r rT s rT c @ s e Zd Zdd� Zdd� ZdS )�TestToLatexBoldc C s: t ddgddgd��}|jdd�}td�}||ks6t�d S ) Nr r r r )r r T)� bold_rowsz� \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule \textbf{0} & 1 & b1 \\ \textbf{1} & 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �test_to_latex_bold_rows� s z'TestToLatexBold.test_to_latex_bold_rowsc C s: t ddgddgd��}|jdd�}td�}||ks6t�d S ) Nr r r r )r r F)rf z� \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �test_to_latex_no_bold_rows� s z*TestToLatexBold.test_to_latex_no_bold_rowsN)rE rF rG rg rh r r r r re � s re c @ s� e Zd Zejdd� �Zejdd� �Zejdd� �Zejdd� �Zejd d � �Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zej�dddddd g�d d � �Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)S )*�TestToLatexCaptionLabelc C s dS )z,Caption for table/tabular LaTeX environment.z/a table in a \texttt{table/tabular} environmentr )r r r r � caption_table� s z%TestToLatexCaptionLabel.caption_tablec C s dS )z@Short caption for testing \caption[short_caption]{full_caption}.za tabler )r r r r � short_caption� s z%TestToLatexCaptionLabel.short_captionc C s dS )z*Label for table/tabular LaTeX environment.ztab:table_tabularr )r r r r � label_table� s z#TestToLatexCaptionLabel.label_tablec C s dS )z(Caption for longtable LaTeX environment.z+a table in a \texttt{longtable} environmentr )r r r r �caption_longtable� s z)TestToLatexCaptionLabel.caption_longtablec C s dS )z&Label for longtable LaTeX environment.z tab:longtabler )r r r r �label_longtable� s z'TestToLatexCaptionLabel.label_longtablec C s$ |j |d�}td�}||ks t�d S )N)�captionan \begin{table} \centering \caption{a table in a \texttt{table/tabular} environment} \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} \end{table} )r r r )r r rj r) r* r r r �test_to_latex_caption_only� s z2TestToLatexCaptionLabel.test_to_latex_caption_onlyc C s$ |j |d�}td�}||ks t�d S )N)�labelaN \begin{table} \centering \label{tab:table_tabular} \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} \end{table} )r r r )r r rl r) r* r r r �test_to_latex_label_only� s z0TestToLatexCaptionLabel.test_to_latex_label_onlyc C s& |j ||d�}td�}||ks"t�d S )N)ro rq a� \begin{table} \centering \caption{a table in a \texttt{table/tabular} environment} \label{tab:table_tabular} \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} \end{table} )r r r )r r rj rl r) r* r r r � test_to_latex_caption_and_label� s z7TestToLatexCaptionLabel.test_to_latex_caption_and_labelc C s( |j ||fd�}td�}||ks$t�d S )N)ro aw \begin{table} \centering \caption[a table]{a table in a \texttt{table/tabular} environment} \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} \end{table} )r r r )r r rj rk r) r* r r r �&test_to_latex_caption_and_shortcaption s z>TestToLatexCaptionLabel.test_to_latex_caption_and_shortcaptionc C s0 d}|j |d�}|j t|�d�}||ks,t�d S )N)zLong-long-captionZShort)ro )r �listr )r r ro Z result_tupleZ result_listr r r �1test_to_latex_caption_and_shortcaption_list_is_ok# s zITestToLatexCaptionLabel.test_to_latex_caption_and_shortcaption_list_is_okc C s* |j ||f|d�}td�}||ks&t�d S )N)ro rq a� \begin{table} \centering \caption[a table]{a table in a \texttt{table/tabular} environment} \label{tab:table_tabular} \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} \end{table} )r r r )r r rj rk rl r) r* r r r �,test_to_latex_caption_shortcaption_and_label) s zDTestToLatexCaptionLabel.test_to_latex_caption_shortcaption_and_label� bad_caption)� full_captionrk Z extra_string)ry rk r )ry rk N)ry )Nc C s< t ddgi�}d}tjt|d�� |j|d� W d Q R X d S )Nr r z9caption must be either a string or a tuple of two strings)r2 )ro )r r4 r5 r6 r )r rx r& r7 r r r � test_to_latex_bad_caption_raisesH s z8TestToLatexCaptionLabel.test_to_latex_bad_caption_raisesc C s$ |j dd�}td�}||ks t�d S )N�xy)ro aA \begin{table} \centering \caption{xy} \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} \end{table} )r r r )r r r) r* r r r � test_to_latex_two_chars_captionY s z7TestToLatexCaptionLabel.test_to_latex_two_chars_captionc C s& |j d|d�}td�}||ks"t�d S )NT)rK ro ay \begin{longtable}{lrl} \caption{a table in a \texttt{longtable} environment}\\ \toprule {} & a & b \\ \midrule \endfirsthead \caption[]{a table in a \texttt{longtable} environment} \\ \toprule {} & a & b \\ \midrule \endhead \midrule \multicolumn{3}{r}{{Continued on next page}} \\ \midrule \endfoot \bottomrule \endlastfoot 0 & 1 & b1 \\ 1 & 2 & b2 \\ \end{longtable} )r r r )r r rm r) r* r r r �$test_to_latex_longtable_caption_onlyo s z<TestToLatexCaptionLabel.test_to_latex_longtable_caption_onlyc C s& |j d|d�}td�}||ks"t�d S )NT)rK rq a \begin{longtable}{lrl} \label{tab:longtable}\\ \toprule {} & a & b \\ \midrule \endfirsthead \toprule {} & a & b \\ \midrule \endhead \midrule \multicolumn{3}{r}{{Continued on next page}} \\ \midrule \endfoot \bottomrule \endlastfoot 0 & 1 & b1 \\ 1 & 2 & b2 \\ \end{longtable} )r r r )r r rn r) r* r r r �"test_to_latex_longtable_label_only� s z:TestToLatexCaptionLabel.test_to_latex_longtable_label_onlyc C s( |j d||d�}td�}||ks$t�d S )NT)rK ro rq a� \begin{longtable}{lrl} \caption{a table in a \texttt{longtable} environment} \label{tab:longtable}\\ \toprule {} & a & b \\ \midrule \endfirsthead \caption[]{a table in a \texttt{longtable} environment} \\ \toprule {} & a & b \\ \midrule \endhead \midrule \multicolumn{3}{r}{{Continued on next page}} \\ \midrule \endfoot \bottomrule \endlastfoot 0 & 1 & b1 \\ 1 & 2 & b2 \\ \end{longtable} )r r r )r r rm rn r) r* r r r �)test_to_latex_longtable_caption_and_label� s zATestToLatexCaptionLabel.test_to_latex_longtable_caption_and_labelc C s, |j d||f|d�}td�}||ks(t�d S )NT)rK ro rq a� \begin{longtable}{lrl} \caption[a table]{a table in a \texttt{longtable} environment} \label{tab:longtable}\\ \toprule {} & a & b \\ \midrule \endfirsthead \caption[]{a table in a \texttt{longtable} environment} \\ \toprule {} & a & b \\ \midrule \endhead \midrule \multicolumn{3}{r}{{Continued on next page}} \\ \midrule \endfoot \bottomrule \endlastfoot 0 & 1 & b1 \\ 1 & 2 & b2 \\ \end{longtable} )r r r )r r rm rk rn r) r* r r r �6test_to_latex_longtable_caption_shortcaption_and_label� s zNTestToLatexCaptionLabel.test_to_latex_longtable_caption_shortcaption_and_labelN)rE rF rG r4 �fixturerj rk rl rm rn rp rr rs rt rv rw rH rI rz r| r} r~ r r� r r r r ri � s. (ri c @ sB e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d � Z d d � Z d S )�TestToLatexEscapec c s, d}d}t |d|di|d|did��V dS )z=Dataframe with special characters for testing chars escaping.r r )zco$e^x$zco^l1N)r )r r r r r r �df_with_symbols s z!TestToLatexEscape.df_with_symbolsc C s$ |j dd�}td�}||ks t�d S )NF)�escapez� \begin{tabular}{lll} \toprule {} & co$e^x$ & co^l1 \\ \midrule a & a & a \\ b & b & b \\ \bottomrule \end{tabular} )r r r )r r� r) r* r r r �test_to_latex_escape_false s z,TestToLatexEscape.test_to_latex_escape_falsec C s |� � }td�}||ks t�d S )Na \begin{tabular}{lll} \toprule {} & co\$e\textasciicircum x\$ & co\textasciicircum l1 \\ \midrule a & a & a \\ b & b & b \\ \bottomrule \end{tabular} )r r r )r r� r) r* r r r � test_to_latex_escape_default s z.TestToLatexEscape.test_to_latex_escape_defaultc C s. t dddg�}|�� }td�}||ks*t�d S )Nza\b\cz^a^b^cz~a~b~cao \begin{tabular}{ll} \toprule {} & 0 \\ \midrule 0 & a\textbackslash b\textbackslash c \\ 1 & \textasciicircum a\textasciicircum b\textasciicircum c \\ 2 & \textasciitilde a\textasciitilde b\textasciitilde c \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r � test_to_latex_special_escape' s z.TestToLatexEscape.test_to_latex_special_escapec C sB ddddddddd d g }t |d �}|�� }td �}||ks>t�d S ) N�&�%�$�#�_�{rR �~�^�\)�dataa� \begin{tabular}{ll} \toprule {} & 0 \\ \midrule 0 & \& \\ 1 & \% \\ 2 & \$ \\ 3 & \# \\ 4 & \_ \\ 5 & \{ \\ 6 & \} \\ 7 & \textasciitilde \\ 8 & \textasciicircum \\ 9 & \textbackslash \\ \bottomrule \end{tabular} )r r r r )r Zspecial_charactersr& r) r* r r r �"test_to_latex_escape_special_chars9 s z4TestToLatexEscape.test_to_latex_escape_special_charsc C s@ t ddgddgd��}|jddgdd �}td �}||ks<t�d S ) Nr r r r )r r z$A$z$B$F)rU r� z� \begin{tabular}{lrl} \toprule {} & $A$ & $B$ \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �;test_to_latex_specified_header_special_chars_without_escapeS s zMTestToLatexEscape.test_to_latex_specified_header_special_chars_without_escapeN) rE rF rG r4 r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�TestToLatexPositionc C s> d}t ddgddgd��}|j|d�}td�}||ks:t�d S ) N�hr r r r )r r )�positiona+ \begin{table}[h] \centering \begin{tabular}{lrl} \toprule {} & a & b \\ \midrule 0 & 1 & b1 \\ 1 & 2 & b2 \\ \bottomrule \end{tabular} \end{table} )r r r r )r � the_positionr& r) r* r r r �test_to_latex_positiong s z*TestToLatexPosition.test_to_latex_positionc C s@ d}t ddgddgd��}|jd|d�}td �}||ks<t�d S ) N�tr r r r )r r T)rK r� a� \begin{longtable}[t]{lrl} \toprule {} & a & b \\ \midrule \endfirsthead \toprule {} & a & b \\ \midrule \endhead \midrule \multicolumn{3}{r}{{Continued on next page}} \\ \midrule \endfoot \bottomrule \endlastfoot 0 & 1 & b1 \\ 1 & 2 & b2 \\ \end{longtable} )r r r r )r r� r& r) r* r r r � test_to_latex_longtable_position| s z4TestToLatexPosition.test_to_latex_longtable_positionN)rE rF rG r� r� r r r r r� f s r� c @ s> e Zd Zdd� Zdd� Zdd� Zej�ddd g�d d � �Z d S ) �TestToLatexFormattersc C s� t tddd�tddd�tddd�gdddgdddgd d d gd ��}d d� dd� dd� dd� dd� d�}|jt|�d�}td�}||ks�t�d S )Ni� r r r/ rO g �?g @g @)r r TF)� datetime64�float�int�objectc S s | � d�S )Nz%Y-%m)�strftime)�xr r r �<lambda>� � zETestToLatexFormatters.test_to_latex_with_formatters.<locals>.<lambda>c S s d| d�d�S )N�[z 4.1f�]r )r� r r r r� � r� c S s d| d��S )N�0xr� r )r� r r r r� � r� c S s d| �d�S )N�-r )r� r r r r� � r� c S s d| � �S )Nzindex: r )r� r r r r� � r� )r� r� r� r� � __index__)� formattersa� \begin{tabular}{llrrl} \toprule {} & datetime64 & float & int & object \\ \midrule index: 0 & 2016-01 & [ 1.0] & 0x1 & -(1, 2)- \\ index: 1 & 2016-02 & [ 2.0] & 0x2 & -True- \\ index: 2 & 2016-03 & [ 3.0] & 0x3 & -False- \\ \bottomrule \end{tabular} )r r r �dictr r )r r& r� r) r* r r r � test_to_latex_with_formatters� s z3TestToLatexFormatters.test_to_latex_with_formattersc C s2 t ddgi�}|jdd�}td�}||ks.t�d S )Nr� g vöE��?z%.3f)r= z� \begin{tabular}{lr} \toprule {} & x \\ \midrule 0 & 0.200 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �3test_to_latex_float_format_no_fixed_width_3decimals� s zITestToLatexFormatters.test_to_latex_float_format_no_fixed_width_3decimalsc C s2 t ddgi�}|jdd�}td�}||ks.t�d S )Nr� g Y@z%.0f)r= z� \begin{tabular}{lr} \toprule {} & x \\ \midrule 0 & 100 \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �1test_to_latex_float_format_no_fixed_width_integer� s zGTestToLatexFormatters.test_to_latex_float_format_no_fixed_width_integer�na_rep�NaNZTedc C sL t ddgdd ggddgd�}|j|djd�}td|� d ��}||ksHt�d S ) Nr\ g��(\��?�Group�Data)�columnsz{:.2f})r� r= z� \begin{tabular}{llr} \toprule {} & Group & Data \\ \midrule 0 & A & 1.22 \\ 1 & A & zB \\ \bottomrule \end{tabular} )r r r> r r )r r� r& r) r* r r r �%test_to_latex_na_rep_and_float_format� s z;TestToLatexFormatters.test_to_latex_na_rep_and_float_formatN) rE rF rG r� r� r� r4 rH rI r� r r r r r� � s &r� c @ s e Zd Zejdd� �Zejdd� �Zdd� Zdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zej�d d d!g�ej�d"d d#g�ej�d$d%gd&gd%d&gg�d'd(� ���Zej�d)d*d+g�d,d-� �Zd.d/� Zd0d1� Zd S )2�TestToLatexMultiindexc c sx t �tdd� td�D ��tdd� td�D ��tdd� td�D ��tdd� td�D ��tdd� td�D ��d��jV d S ) z7Multiindex dataframe for testing multirow LaTeX macros.c S s i | ] }||�qS r r )�.0r� r r r � <dictcomp> s z:TestToLatexMultiindex.multiindex_frame.<locals>.<dictcomp>rP c S s i | ]}|d |�qS )rP r )r� r� r r r r� s c S s i | ] }||�qS r r )r� r� r r r r� s c S s i | ]}|d |�qS )rP r )r� r� r r r r� s c S s i | ] }||�qS r r )r� r� r r r r� s ))�c1r )r� r )�c2r )r� r )�c3r N)r � from_dictr �range�T)r r r r �multiindex_frame� s z&TestToLatexMultiindex.multiindex_framec c s` t dd� td�D �dd� td�D �dd� td�D �dd� td�D �dd� td�D �d��V d S ) z;Multicolumn dataframe for testing multicolumn LaTeX macros.c S s i | ] }||�qS r r )r� r� r r r r� s z;TestToLatexMultiindex.multicolumn_frame.<locals>.<dictcomp>r/ c S s i | ]}|d |�qS )r/ r )r� r� r r r r� s c S s i | ] }||�qS r r )r� r� r r r r� s c S s i | ]}|d |�qS )r/ r )r� r� r r r r� s c S s i | ] }||�qS r r )r� r� r r r r� s ))r� r )r� r )r� r )r� r )r� r N)r r� )r r r r �multicolumn_frame s z'TestToLatexMultiindex.multicolumn_framec C sP t dgdgdgdgd��}|�ddg�}|jdd gd �}td �}||ksLt�d S ) Nr r r rO )r r r1 �dr r �r1�r2)rU z� \begin{tabular}{llrr} \toprule & & r1 & r2 \\ a & b & & \\ \midrule 0 & 1 & 2 & 3 \\ \bottomrule \end{tabular} )r � set_indexr r r )r r& �observedr* r r r � test_to_latex_multindex_header s z4TestToLatexMultiindex.test_to_latex_multindex_headerc C sJ t jjddggdgd�}td|td�d�}|�� }td�}||ksFt�d S ) Nr r � )�names���rP )r, r� a! \begin{tabular}{lrrrr} \toprule & 0 & 1 & 2 & 3 \\ {} & & & & \\ \midrule 1 & -1 & -1 & -1 & -1 \\ 2 & -1 & -1 & -1 & -1 \\ \bottomrule \end{tabular} )�pd� MultiIndex� from_productr r� r r r )r �mir& r� r* r r r �#test_to_latex_multiindex_empty_name+ s z9TestToLatexMultiindex.test_to_latex_multiindex_empty_namec C s. t ddgi�}|�� }td�}||ks*t�d S )N)r� �yr z� \begin{tabular}{ll} \toprule {} & x \\ {} & y \\ \midrule 0 & a \\ \bottomrule \end{tabular} )r r r r )r r& r) r* r r r �'test_to_latex_multiindex_column_tabular? s z=TestToLatexMultiindex.test_to_latex_multiindex_column_tabularc C s0 t ddgi�j}|�� }td�}||ks,t�d S )N)r� r� r z� \begin{tabular}{lll} \toprule & & 0 \\ \midrule x & y & a \\ \bottomrule \end{tabular} )r r� r r r )r r& r) r* r r r �&test_to_latex_multiindex_small_tabularP s z<TestToLatexMultiindex.test_to_latex_multiindex_small_tabularc C s |� � }td�}||ks t�d S )Na� \begin{tabular}{llrrrr} \toprule & & 0 & 1 & 2 & 3 \\ \midrule c1 & 0 & 0 & 1 & 2 & 3 \\ & 1 & 4 & 5 & 6 & 7 \\ c2 & 0 & 0 & 1 & 2 & 3 \\ & 1 & 4 & 5 & 6 & 7 \\ c3 & 0 & 0 & 1 & 2 & 3 \\ \bottomrule \end{tabular} )r r r )r r� r) r* r r r � test_to_latex_multiindex_tabular` s z6TestToLatexMultiindex.test_to_latex_multiindex_tabularc C s2 |j }ddg|j_|�� }td�}||ks.t�d S )Nr r a� \begin{tabular}{lrrrrr} \toprule a & \multicolumn{2}{l}{c1} & \multicolumn{2}{l}{c2} & c3 \\ b & 0 & 1 & 0 & 1 & 0 \\ \midrule 0 & 0 & 4 & 0 & 4 & 0 \\ 1 & 1 & 5 & 1 & 5 & 1 \\ 2 & 2 & 6 & 2 & 6 & 2 \\ 3 & 3 & 7 & 3 & 7 & 3 \\ \bottomrule \end{tabular} )r� r� r� r r r )r r� r& r) r* r r r �!test_to_latex_multicolumn_tabulars s z7TestToLatexMultiindex.test_to_latex_multicolumn_tabularc C sN t ddddgtd�ddddgd��}|�dd g��� }td �}||ksJt�d S ) Nr r �ababr rO rP )r r r1 r r a& \begin{tabular}{llr} \toprule & & c \\ a & b & \\ \midrule 0 & a & 1 \\ & b & 2 \\ 1 & a & 3 \\ & b & 4 \\ \bottomrule \end{tabular} )r ru r� r r r )r r& r) r* r r r �$test_to_latex_index_has_name_tabular� s $ z:TestToLatexMultiindex.test_to_latex_index_has_name_tabularc C sN t ddddgtd�ddddgd��}|�d��� �� }td �}||ksJt�d S ) Nr r r� r rO rP )r r r1 r a \begin{tabular}{lrrrrrrrr} \toprule {} & \multicolumn{8}{l}{c} \\ {} & count & mean & std & min & 25\% & 50\% & 75\% & max \\ a & & & & & & & & \\ \midrule 0 & 2.0 & 1.5 & 0.707107 & 1.0 & 1.25 & 1.5 & 1.75 & 2.0 \\ 1 & 2.0 & 3.5 & 0.707107 & 3.0 & 3.25 & 3.5 & 3.75 & 4.0 \\ \bottomrule \end{tabular} )r ru �groupbyZdescriber r r )r r& r) r* r r r � test_to_latex_groupby_tabular� s $ z3TestToLatexMultiindex.test_to_latex_groupby_tabularc C s: t tj�ddg�dgd�}|�� }td�}||ks6t�d S )N)r\ r1 )r] r1 �col)r, r� z� \begin{tabular}{lll} \toprule & & col \\ \midrule A & c & NaN \\ B & c & NaN \\ \bottomrule \end{tabular} )r r� r� � from_tuplesr r r )r r& r) r* r r r �#test_to_latex_multiindex_dupe_level� s z9TestToLatexMultiindex.test_to_latex_multiindex_dupe_levelc C s |� � }td�}||ks t�d S )Na� \begin{tabular}{lrrrrr} \toprule {} & \multicolumn{2}{l}{c1} & \multicolumn{2}{l}{c2} & c3 \\ {} & 0 & 1 & 0 & 1 & 0 \\ \midrule 0 & 0 & 5 & 0 & 5 & 0 \\ 1 & 1 & 6 & 1 & 6 & 1 \\ 2 & 2 & 7 & 2 & 7 & 2 \\ 3 & 3 & 8 & 3 & 8 & 3 \\ 4 & 4 & 9 & 4 & 9 & 4 \\ \bottomrule \end{tabular} )r r r )r r� r) r* r r r �!test_to_latex_multicolumn_default� s z7TestToLatexMultiindex.test_to_latex_multicolumn_defaultc C s$ |j dd�}td�}||ks t�d S )NF)� multicolumna� \begin{tabular}{lrrrrr} \toprule {} & c1 & & c2 & & c3 \\ {} & 0 & 1 & 0 & 1 & 0 \\ \midrule 0 & 0 & 5 & 0 & 5 & 0 \\ 1 & 1 & 6 & 1 & 6 & 1 \\ 2 & 2 & 7 & 2 & 7 & 2 \\ 3 & 3 & 8 & 3 & 8 & 3 \\ 4 & 4 & 9 & 4 & 9 & 4 \\ \bottomrule \end{tabular} )r r r )r r� r) r* r r r � test_to_latex_multicolumn_false� s z5TestToLatexMultiindex.test_to_latex_multicolumn_falsec C s& |j jdd�}td�}||ks"t�d S )NT)�multirowa� \begin{tabular}{llrrrrr} \toprule & & 0 & 1 & 2 & 3 & 4 \\ \midrule \multirow{2}{*}{c1} & 0 & 0 & 1 & 2 & 3 & 4 \\ & 1 & 5 & 6 & 7 & 8 & 9 \\ \cline{1-7} \multirow{2}{*}{c2} & 0 & 0 & 1 & 2 & 3 & 4 \\ & 1 & 5 & 6 & 7 & 8 & 9 \\ \cline{1-7} c3 & 0 & 0 & 1 & 2 & 3 & 4 \\ \bottomrule \end{tabular} )r� r r r )r r� r) r* r r r �test_to_latex_multirow_true� s z1TestToLatexMultiindex.test_to_latex_multirow_truec C s4 |j j|_|j jdddd�}td�}||ks0t�d S )NTr1 )r� r� �multicolumn_formataG \begin{tabular}{llrrrrr} \toprule & & \multicolumn{2}{c}{c1} & \multicolumn{2}{c}{c2} & c3 \\ & & 0 & 1 & 0 & 1 & 0 \\ \midrule \multirow{2}{*}{c1} & 0 & 0 & 1 & 2 & 3 & 4 \\ & 1 & 5 & 6 & 7 & 8 & 9 \\ \cline{1-7} \multirow{2}{*}{c2} & 0 & 0 & 1 & 2 & 3 & 4 \\ & 1 & 5 & 6 & 7 & 8 & 9 \\ \cline{1-7} c3 & 0 & 0 & 1 & 2 & 3 & 4 \\ \bottomrule \end{tabular} )r� r, r r r )r r� r) r* r r r �1test_to_latex_multicolumnrow_with_multicol_format s zGTestToLatexMultiindex.test_to_latex_multicolumnrow_with_multicol_format�name0NZnamed0�name1Znamed1�axesr r c s� ||g}t j�ddgddgg�}td|�� |�� d�}x� D ]}||j| _q<W tdd� |D ��}d � kr�t|�r�|d � d |d � d �nd } t|�r�d� kr�d nd�� �fdd�|D �} |� � } dtt | �| g � } | | ks�t �d S )Nr r rO rP r� )r, r� c s s | ]}|p d V qdS )z{}Nr )r� �nr r r � <genexpr>/ s zGTestToLatexMultiindex.test_to_latex_multiindex_names.<locals>.<genexpr>r z & z & & & & \\ r� z{}� c s$ g | ] }t |�r d � kr |n��qS )r )�bool)r� r� )r� � placeholderr r � <listcomp>6 s zHTestToLatexMultiindex.test_to_latex_multiindex_names.<locals>.<listcomp>a \begin{tabular}{llrrrr} \toprule & %s & \multicolumn{2}{l}{1} & \multicolumn{2}{l}{2} \\ & %s & 3 & 4 & 3 & 4 \\ %s\midrule 1 & 3 & -1 & -1 & -1 & -1 \\ & 4 & -1 & -1 & -1 & -1 \\ 2 & 3 & -1 & -1 & -1 & -1 \\ & 4 & -1 & -1 & -1 & -1 \\ \bottomrule \end{tabular} ) r� r� r� r �copyr� r� �tuple�anyr ru r ) r r� r� r� r� r� r& �idxZ idx_namesZ idx_names_rowZ col_namesr� r* r )r� r� r � test_to_latex_multiindex_names$ s ( z4TestToLatexMultiindex.test_to_latex_multiindex_names�one_rowTFc C sj t d dgddgddgd��}|r,|jdg }|�dd g��� }td �}|sR|d 7 }|d 7 }||ksft�d S ) Nr r rO rP r/ )r r r1 r r r z� \begin{tabular}{llr} \toprule & & c \\ a & b & \\ \midrule NaN & 2 & 4 \\ z1.0 & 3 & 5 \\ z\bottomrule \end{tabular} )r Zilocr� r r r )r r� r& r� r* r r r � test_to_latex_multiindex_nansH s z3TestToLatexMultiindex.test_to_latex_multiindex_nansc C s> t dddggd ��ddg�}|�� }td�}||ks:t�d S )Nr r rO r z� \begin{tabular}{llr} \toprule & & 2 \\ 0 & 1 & \\ \midrule 1 & 2 & 3 \\ & 2 & 3 \\ \bottomrule \end{tabular} )r r� r r r )r r& r) r* r r r � test_to_latex_non_string_indexa s z4TestToLatexMultiindex.test_to_latex_non_string_indexc C sZ t jjddgdddgddggddd gd �}t|d �}|jd d d�}td�}||ksVt�d S )Ng g �?g @g @�0�1�iZval0Zval1)r� )r, TF)r� r� a: \begin{tabular}{lll} \toprule & & \\ i & val0 & val1 \\ \midrule \multirow{6}{*}{0.0} & \multirow{2}{*}{3.0} & 0 \\ & & 1 \\ \cline{2-3} & \multirow{2}{*}{2.0} & 0 \\ & & 1 \\ \cline{2-3} & \multirow{2}{*}{1.0} & 0 \\ & & 1 \\ \cline{1-3} \cline{2-3} \multirow{6}{*}{1.0} & \multirow{2}{*}{3.0} & 0 \\ & & 1 \\ \cline{2-3} & \multirow{2}{*}{2.0} & 0 \\ & & 1 \\ \cline{2-3} & \multirow{2}{*}{1.0} & 0 \\ & & 1 \\ \bottomrule \end{tabular} )r� r� r� r r r r )r r� r& r) r* r r r �!test_to_latex_multiindex_multirowt s $ z7TestToLatexMultiindex.test_to_latex_multiindex_multirow)rE rF rG r4 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rH rI r� r� r� r� r r r r r� � s* ("r� c @ s@ e Zd Zejdd� �Zejdd� �Zdd� Zdd� Zd d � Z d S ) �TestTableBuilderc C s t ddgddgd��S )Nr r r r )r r )r )r r r r � dataframe� s zTestTableBuilder.dataframec C s t t|�d�S )N)� formatter)r r )r r� r r r � table_builder� s z TestTableBuilder.table_builderc C s |j dd�}t|t�st�d S )NrU )�over)�_create_row_iterator� isinstancer r )r r� �iteratorr r r �test_create_row_iterator� s z)TestTableBuilder.test_create_row_iteratorc C s |j dd�}t|t�st�d S )N�body)r� )r� r� r r )r r� r� r r r �test_create_body_iterator� s z*TestTableBuilder.test_create_body_iteratorc C s* t jtdd�� |jdd� W d Q R X d S )Nz!must be either 'header' or 'body')r2 z SOMETHING BAD)r� )r4 r5 r6 r� )r r� r r r �#test_create_body_wrong_kwarg_raises� s z4TestTableBuilder.test_create_body_wrong_kwarg_raisesN) rE rF rG r4 r� r� r� r� r� r r r r r r� � s r� c @ s` e Zd Zej�ddddg�dd� �Zej�dddd g�d d � �Zej�dd d dg�dd� �ZdS )�TestRowStringConverterzrow_num, expected)r z {} & Design & ratio & xy \\)r z 0 & 1 & 4 & 10 \\)r z 1 & 2 & 5 & 11 \\c C sL t dddgdddgddd gd ��}tt|d d �d �}|j|d�|ksHt�d S )Nr r rO rP r/ rQ � � � )ZDesign�ratior{ T)r� )r� )�row_num)r r r � get_strrowr )r r r* r& �row_string_converterr r r �%test_get_strrow_normal_without_escape� s "z<TestRowStringConverter.test_get_strrow_normal_without_escape)r z'{} & Design \# & ratio, \% & x\&y \\)r z#0 & 1 & 4 & 10 \\)r z#1 & 2 & 5 & 11 \\c C sL t dddgdddgddd gd ��}tt|d d �d �}|j|d�|ksHt�d S )Nr r rO rP r/ rQ r r r )zDesign #zratio, %zx&yT)r� )r� )r )r r r r r )r r r* r& r r r r �"test_get_strrow_normal_with_escape� s z9TestRowStringConverter.test_get_strrow_normal_with_escape)r z<{} & \multicolumn{2}{r}{c1} & \multicolumn{2}{r}{c2} & c3 \\)r z {} & 0 & 1 & 0 & 1 & 0 \\)r z 0 & 0 & 5 & 0 & 5 & 0 \\c C s� t dd� td�D �dd� td�D �dd� td�D �dd� td�D �dd� td�D �d��}tt|�d d d d �}|j|d �|ks�t�d S ) Nc S s i | ] }||�qS r r )r� r� r r r r� � s zPTestRowStringConverter.test_get_strrow_multindex_multicolumn.<locals>.<dictcomp>r/ c S s i | ]}|d |�qS )r/ r )r� r� r r r r� � s c S s i | ] }||�qS r r )r� r� r r r r� � s c S s i | ]}|d |�qS )r/ r )r� r� r r r r� � s c S s i | ] }||�qS r r )r� r� r r r r� � s ))r� r )r� r )r� r )r� r )r� r Tr$ )r� r� r� r� )r )r r� r r r r )r r r* r& r r r r �%test_get_strrow_multindex_multicolumn� s z<TestRowStringConverter.test_get_strrow_multindex_multicolumnN) rE rF rG r4 rH rI r r r r r r r r � s r )!r% r �textwrapr r4 Zpandasr� r r Zpandas._testingZ_testingr Zpandas.io.formats.formatr Zpandas.io.formats.latexr r r r r r� r r rJ rT re ri r� r� r� r� r� r r r r r �<module> s6 !Wl& Qg5b !
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка