Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/parser/__pycache__/test_header.cpython-37.pyc
Назад
B �5�g�? � @ s d Z ddlmZ ddlmZ ddlZddlZddlm Z ddl m Z m Z m Z ddlmZ dd� Zd d � Zej�d d d dgddgg�dd� �Zej�d ddg�dd� �Zdd� Zdd� Zdd� Zd d � Zej�d d d d!gid"fdd#gd d!gd$�d%fdd#gd d!gd&�d'fg�d(d)� �Zed*d+d,g�Z ej�d-d dd#gid.d/d0d1d2d3d4gd5�d.e d6d7�e d6d8�e d6d9�e d:d;�e d<d=�e d<d>�gd5�g�d?d@� �Z ej�d-d dd#gid d/d0d1d2d3d4gd5�d e d6d7�e d6d8�e d6d9�e d:d;�e d<d=�e d<d>�gd5�g�dAdB� �Z ej�d-d dd#gid d/d0d1d2d3d4gd5�d e d6d7�e d6d8�e d6d9�e d:d;�e d<d=�e d<d>�gd5�g�dCdD� �Z dEdF� Z dGdH� Z!dIdJ� Z"dKdL� Z#ej�dMdNdOg�dPdQ� �Z$ej�d-i d dig�dRdS� �Z%ej�dTi dd#d d.dgfdUdVidWdXdYdZd[gfd\d d!d]d^d_gid d!d]d^d_gfg�d`da� �Z&ej�d d6d:gdbg�dcdd� �Z'dedf� Z(ej�dgdhe ddidjdkgge �)dldmdndog�dp�fdqe ddidjdkgge �)dldmdrdog�dp�fdse ddidjdkdkgge �)dldmdrdodtg�dp�fg�dudv� �Z*ej�d ddgg�ej�dwddxdygdydxgdydzgg�d{d|� ��Z+d}d~� Z,dS )zx Tests that the file header is properly handled or inferred during parsing for all of the parsers defined in parsers.py � )� namedtuple)�StringION)� ParserError)� DataFrame�Index� MultiIndexc C s> | }d}t jt|d�� td�}|j|dgd� W d Q R X d S )Nzbut only \d+ lines in file)�matchz,,� )�header)�pytest�raises� ValueErrorr �read_csv)� all_parsers�parser�msg�s� r �X/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/parser/test_header.py�test_read_with_bad_header s r c C s8 | }d}t jtdd�� |jt|�dd� W d Q R X d S )Nz$1,2,3,4,5 6,7,8,9,10 11,12,13,14,15 zUPassing negative integer to header is invalid. For no header, use header=None instead)r ���)r )r r r r r )r r �datar r r �test_negative_header s r r r � � ���c C s8 | }d}t jtdd�� |jt|�|d� W d Q R X d S )Nz<1,2,3,4,5 6,7,8,9,10 11,12,13,14,15 z8cannot specify multi-index header with negative integers)r )r )r r r r r )r r r r r r r � test_negative_multi_index_header. s r TFc C s< | }d}d}t jt|d�� |jt|�|d� W d Q R X d S )NzMyColumn a b a bz#Passing a bool to header is invalid)r )r )r r � TypeErrorr r )r r r r r r r r �test_bool_header_arg< s r c C sf | }d}|j t|�dd d�}tdddddgd d d d d gdddddggdddddgd�}t�||� d S )Nz$1,2,3,4,5 6,7,8,9,10 11,12,13,14,15 �Field)�prefixr � r � r � � � � � r � � � � � ZField0ZField1ZField2ZField3ZField4)�columns)r r r �tm�assert_frame_equal)r r r �result�expectedr r r �test_no_header_prefixK s &r2 c C sf | }d}dddg}|j t|�|d�}tdddgd d d gd d dggdddgdddgd�}t�||� d S )Nz foo,1,2,3 bar,4,5,6 baz,7,8,9 �A�B�C)�namesr! r r" r r# r$ r% r& r' �foo�bar�baz)�indexr- )r r r r. r/ )r r r r6 r0 r1 r r r �test_header_with_index_colY s r; c C sD | }d}d}|j t|�ddd�}|j t|�ddd�}t�||� d S )Nzggot,to,ignore,this,line got,to,ignore,this,line index,A,B,C,D foo,2,3,4,5 bar,7,8,9,10 baz,12,13,14,15 z7index,A,B,C,D foo,2,3,4,5 bar,7,8,9,10 baz,12,13,14,15 r r )r � index_col)r r r. r/ )r r r Zdata2r0 r1 r r r �test_header_not_first_linej s r= c C sJ | }t jddddd�}d}|jt|�ddddgddgd �}t �||� d S ) Nr# r" r r )Z r_idx_nlevelsZ c_idx_nlevelsa C0,,C_l0_g0,C_l0_g1,C_l0_g2 C1,,C_l1_g0,C_l1_g1,C_l1_g2 C2,,C_l2_g0,C_l2_g1,C_l2_g2 C3,,C_l3_g0,C_l3_g1,C_l3_g2 R0,R1,,, R_l0_g0,R_l1_g0,R0C0,R0C1,R0C2 R_l0_g1,R_l1_g1,R1C0,R1C1,R1C2 R_l0_g2,R_l1_g2,R2C0,R2C1,R2C2 R_l0_g3,R_l1_g3,R3C0,R3C1,R3C2 R_l0_g4,R_l1_g4,R4C0,R4C1,R4C2 r r! )r r< )r. ZmakeCustomDataframer r r/ )r r r1 r r0 r r r �test_header_multi_index~ s r> z kwargs,msgr< r7 r8 zLindex_col must only contain row numbers when specifying a multi-index headerr! )r< r6 z9cannot specify names when specifying a multi-index header)r< Zusecolsz;cannot specify usecols when specifying a multi-index headerc C sH d}| }t jt|d��( |jt|�fdddddgi|�� W d Q R X d S )Na C0,,C_l0_g0,C_l0_g1,C_l0_g2 C1,,C_l1_g0,C_l1_g1,C_l1_g2 C2,,C_l2_g0,C_l2_g1,C_l2_g2 C3,,C_l3_g0,C_l3_g1,C_l3_g2 R0,R1,,, R_l0_g0,R_l1_g0,R0C0,R0C1,R0C2 R_l0_g1,R_l1_g1,R1C0,R1C1,R1C2 R_l0_g2,R_l1_g2,R2C0,R2C1,R2C2 R_l0_g3,R_l1_g3,R3C0,R3C1,R3C2 R_l0_g4,R_l1_g4,R4C0,R4C1,R4C2 )r r r r! r r" )r r r r r )r �kwargsr r r r r r � test_header_multi_index_invalid� s "r@ � _TestTuple�first�secondr? r" )�a�q)rD �r)rD r )�b�t)�c�u)rI �v)Zskiprowsr6 rD rE rF r rG rH rI rJ rK c C sr | }t ddddddgddd d d d ggd dgt�ddddddg�d�}d}|jt|�fddi|��}t�||� d S )Nr! r r" r r# r$ r% r& r' r r( r) �one�two)rD rE )rD rF )rD r )rG rH )rI rJ )rI rK )r: r- zC,a,a,a,b,c,c ,q,r,s,t,u,v ,,,,,, one,1,2,3,4,5,6 two,7,8,9,10,11,12r< r )r r � from_tuplesr r r. r/ )r r? r r1 r r0 r r r �&test_header_multi_index_common_format1� s rO c C sr | }t ddddddgddd d d d ggd dgt�ddddddg�d�}d}|jt|�fddi|��}t�||� d S )Nr! r r" r r# r$ r% r& r' r r( r) rL rM )rD rE )rD rF )rD r )rG rH )rI rJ )rI rK )r: r- z<,a,a,a,b,c,c ,q,r,s,t,u,v one,1,2,3,4,5,6 two,7,8,9,10,11,12r< r )r r rN r r r. r/ )r r? r r1 r r0 r r r �&test_header_multi_index_common_format2� s rP c C s~ | }t ddddddgddd d d d ggd dgt�ddddddg�d�}|jdd�}d}|jt|�fdd i|��}t�||� d S )Nr! r r" r r# r$ r% r& r' r r( r) rL rM )rD rE )rD rF )rD r )rG rH )rI rJ )rI rK )r: r- T)Zdropz2a,a,a,b,c,c q,r,s,t,u,v 1,2,3,4,5,6 7,8,9,10,11,12r< )r r rN Z reset_indexr r r. r/ )r r? r r1 r r0 r r r �&test_header_multi_index_common_format3 s rQ c C s� | }t tjdddddgdddd d ggd d �td dg�tdddgdddddggddd ddgdd dddggddgd�d�}d}|jt|�dd gdd �}t�||� d S ) Nr r" r r# r$ r& r' r r( r) �int64)�dtyper! r% rD rG rI rF r rH rJ rK r rE )�levels�codesr6 )r: r- z2a,a,a,b,c,c q,r,s,t,u,v 1,2,3,4,5,6 7,8,9,10,11,12)r r< ) r �np�arrayr r r r r. r/ )r r r1 r r0 r r r �0test_header_multi_index_common_format_malformed1K s $ rX c C s� | }t tjdddddgdddd d ggd d �td dg�tdddgdddddggddd ddgdd dddggd dgd�d�}d}|jt|�dd gdd �}t�||� d S ) Nr r" r r# r$ r& r' r r( r) rR )rS r! r% rD rG rI rF r rH rJ rK r rE )rT rU r6 )r: r- z1,a,a,b,c,c q,r,s,t,u,v 1,2,3,4,5,6 7,8,9,10,11,12)r r< ) r rV rW r r r r r. r/ )r r r1 r r0 r r r �0test_header_multi_index_common_format_malformed2_ s $ rY c C s� | }t tjddddgddddggd d �td d gd dggdd gdd ggd�tdddgddddggdd d d gdd d dggd dgd�d�}d}|jt|�dd gdd gd �}t�||� d S ) Nr" r r# r$ r' r r( r) rR )rS r! r% r r& r )rT rU rD rG rI r rH rJ rK rE )rT rU r6 )r: r- z1,a,a,b,c,c q,r,s,t,u,v 1,2,3,4,5,6 7,8,9,10,11,12)r r< )r rV rW r r r r. r/ )r r r1 r r0 r r r �0test_header_multi_index_common_format_malformed3t s " rZ c C s^ | }d d gddgddgg}t �ddg�}t||d�}d}|jt|�d dgd �}t�||� d S ) Nr! r r" r )rD r3 )rG r4 )r- za,b A,B , 1,2 3,4r )r )r rN r r r r. r/ )r r r r- r1 r0 r r r �"test_header_multi_index_blank_line� s r[ z data,header)z 1,2,3 4,5,6N)zfoo,bar,baz 1,2,3 4,5,6r c C sF | }|j td�dddgd�}|j t|�dddg|d�}t�||� d S )Nz 1,2,3 4,5,6rD rG rI )r6 )r6 r )r r r. r/ )r r r r r1 r0 r r r �!test_header_names_backward_compat� s r\ c C s6 | }t dddgd�}|jtd�f|�}t�||� d S )NrD rG rI )r- za,b,c)r r r r. r/ )r r? r r1 r0 r r r � test_read_only_header_no_rows� s r] z kwargs,namesr �XZX0ZX1ZX2ZX3ZX4r6 r9 ZquuxZpandac C sb | }d}t dddddgddd d d gd d dddgg|d�}|jt|�fdd i|��}t�||� d S )Nz$1,2,3,4,5 6,7,8,9,10 11,12,13,14,15 r! r r" r r# r$ r% r& r' r r( r) r* r+ r, )r- r )r r r r. r/ )r r? r6 r r r1 r0 r r r �test_no_header� s .r_ Z string_headerc C s< d}d}| }t jt|d�� |jt|�|d� W d Q R X d S )Nz*header must be integer or list of integersz1,2 3,4)r )r )r r r r r )r r r r r r r r �test_non_int_header� s r` c C sH d}| }t ddgddgddgd��}|jt|�dgd�}t�||� d S )Nza,b,c 0,1,2 1,2,3r r! r r" )rD rG rI )r )r r r r. r/ )r r r r1 r0 r r r �test_singleton_header� s ra z data,expectedz#A,A,A,B one,one,one,two 0,40,34,0.1�( �" g�������?)r3 rL )r3 zone.1)r3 zone.2)r4 rM )r- z%A,A,A,B one,one,one.1,two 0,40,34,0.1)r3 zone.1.1z/A,A,A,B,B one,one,one.1,two,two 0,40,34,0.1,0.1)r4 ztwo.1c C s* | }|j t|�ddgd�}t�||� d S )Nr r! )r )r r r. r/ )r r r1 r r0 r r r �test_mangles_multi_index� s (rd r- � ZUnnamedZ NotUnnamedc C s | }ddg}|d kr,d� |p"ddg�d }n d� dg|p>ddg �d }|d kr�d}tjt|d�� |jt|�||d � W d Q R X n�|jt|�||d �}g }x>t|�D ]2\} } | s�d |d kr�| n| d � d �} |�| � q�W t� t |d d g��}t ddgddgg|d�} t � || � d S )Nr r! �,re z 0,1 2,3 4,5 z ,0,1 0,2,3 1,4,5 zGPassed header=\[0,1\] are too many rows for this multi_index of columns)r )r r< z Unnamed: Z_level_0�0�1r r" r r# )r- )�joinr r r r r � enumerate�appendr rN �zipr r. r/ ) r r< r- r r r r r0 Z exp_columns�i�colr1 r r r �test_multi_index_unnamed s$ ro c C s� | }d}d}t �dddddg�}tdd d d d gdd d d d gg|d�}|jt|�ddgd�}t�||jd d� � |jt|�ddgd�}t�||� d S )NzFMale, Male, Male, Female, Female R, R, L, R, R .86, .67, .88, .78, .81z^Male, Male, Male, Female, Female R, R, L, R, R .86, .67, .88, .78, .81 .86, .67, .88, .78, .82)ZMale�R)z Malez R)z Malez L)z Femalez R)z Femalez R.1g��Q� ��?gq= ףp�?g)\��(�?g�(\���?g�Q� ���?g= ףp=�?)r- r r! )r )r rN r r r r. r/ Ziloc)r r �s1�s2�mir1 Zdf1Zdf2r r r � test_read_csv_multiindex_columns/ s "rt )-�__doc__� collectionsr �ior �numpyrV r Z pandas.errorsr Zpandasr r r Zpandas._testingZ_testingr. r r �markZ parametrizer r r2 r; r= r> r@ rA rO rP rQ rX rY rZ r[ r\ r] r_ r` ra rN rd ro rt r r r r �<module> s� $ &)
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка