Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/io/parser/usecols/__pycache__/test_usecols_basic.cpython-37.pyc
Назад
B �5�g�, � @ s� d Z ddlmZ ddlZddlZddlmZmZ ddl m Z dZ dZ dd� Zej�d d d g�d d � �Zdd� Zej�dddgddgfdddgddgfg�dd� �Zdd� Zdd� Zd d � Zej�d d d g�d!d"� �Zej�d#ddg�ej�d ddgddgg�d$d%� ��Zd&d'� Zd(d)� Zd*d+� Zd,d-� Z ej�d.ddged/d0gd1d2ggd3d4gd5�fd4d6ged0d7gd2d8ggd4d6gd5�fg�d9d:� �Z d;d<� Z d=d>� Z ej�d.d?d@� edAdBdCdD�dEddFdD�ddddD�dG��fdHd@� e� fg�dIdJ� �Z ej�d ddgdKd@� g�dLdM� �Z!ej�dNdOdddgdPdiedQdRdSgdQdRdSgdTdUdVgg�fdWdXdYdZgi edd[dddd\gdd]dddd^gd[d\d]d[d[dFgd_��fg�d`da� �Z"ej�dbddddcgi edd\gdd^gd[dFgd]dEgdd��dfddddegi de �#df�fdddegi de �#df�fdddedggi de �#dh�fdddXdYdZdigdj�edd\gdd^gd[dFgd]dEgdk��dfdXdYdZdegddXdYdZdigdj�de �#df�fdXdYdegdldXdYdZdigide �#df�fg�dmdn� �Z$ej�d dXdZgddgg�dodp� �Z%ej�dldddgg�dqdr� �Z&dS )sz] Tests the usecols functionality during parsing for all of the parsers defined in parsers.py � )�StringION)� DataFrame�Indexz['usecols' must either be list-like of all strings, all unicode, all integers or a callable.zAUsecols do not match columns, columns expected but not found: {0}c C sB d}dddg}| }t jttd�� |jt|�|d� W d Q R X d S )Nz<a,b,c 1000,2000,3000 4000,5000,6000 r �b� )�match)�usecols)�pytest�raises� ValueError�_msg_validate_usecols_arg�read_csvr )� all_parsers�datar �parser� r �g/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/io/parser/usecols/test_usecols_basic.py�!test_raise_on_mixed_dtype_usecols s r r )� r )r �cc C sR d}| }|j t|�|d�}tddgddgddgd d ggd d gd �}t�||� d S )Nz a,b,c 1,2,3 4,5,6 7,8,9 10,11,12)r r � � � � � � � r r )�columns)r r r �tm�assert_frame_equal)r r r r �result�expectedr r r � test_usecols' s (r" c C s^ d}| }ddg}|j t|�|ddgdd�}tddgd d gd d gd dgg|d�}t�||� d S )Nz a,b,c 1,2,3 4,5,6 7,8,9 10,11,12�foo�barr r r )�namesr �headerr r r r r r r )r )r r r r r )r r r r% r r! r r r �test_usecols_with_names6 s $r' z names,usecolsr r r r �ac C sV d}| }|j t|�|d |d�}tddgddgddgd d ggd d gd �}t�||� d S )Nz1,2,3 4,5,6 7,8,9 10,11,12)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 � test_usecols_relative_to_namesE s (r) c C s^ d}| }|j t|�ddgd ddgd�}tddgdd gd d gd d ggddgd�}t�||� d S )Nz1,2,3 4,5,6 7,8,9 10,11,12r( r r r )r% r& r r � r � r � r )r )r r r r r )r r r r r! r r r � test_usecols_relative_to_names2U s (r- c C sF d}| }d}t jt|d��" |jt|�ddgd dgd� W d Q R X d S )Nz1,2,3 4,5,6 7,8,9 10,11,12zHNumber of passed names did not match number of header fields in the file)r r( r r )r% r& r )r r r r r )r r r �msgr r r �!test_usecols_name_length_conflicte s r/ c C s8 | }d}t jttd�� |jt|�dd� W d Q R X d S )Nz/foo, bar, baz 1000, 2000, 3000 4000, 5000, 6000)r r# )r )r r r r r r )r r r r r r �test_usecols_single_stringr s r0 r za,b,c,d 1,2,3,4 5,6,7,8za,b,c,d 1,2,3,4, 5,6,7,8,c C sN | }dddg}t ddgddgdd gd ��}|jt|�|d d �}t�||� d S ) Nr( r �dr r r r+ r* r )r( r r1 F)r � index_col)r r r r r )r r r r r! r r r r � test_usecols_index_col_false} s r3 r2 c C sL | }d}t dddgitddgdd�d�}|jt|�||d �}t�||� d S ) Nza,b,c,d A,a,1,one B,b,2,twor r r r( r )�name)�index)r r2 )r r r r r r )r r r2 r r r! r r r r � test_usecols_index_col_conflict� s r6 c C s\ | }d}t ddgddgdd��}|�ddg�}|jt|�ddd gddgd �}t�||� d S ) Nza,b,c,d A,a,1,one B,b,2,twor( r r r )�one�two)r r r1 r r1 )r r2 )r Z set_indexr r r r )r r r r! r r r r � test_usecols_index_col_conflict2� s r9 c C sL | }d}|j t|�ddgd�}tddgddgd �d d gd �}t�||� d S ) Nz%a,b,c 4,apple,bat,5.7 8,orange,cow,10r( r )r �apple�orange�bat�cow)r( r r* r )r5 )r r r r r )r r r r r! r r r � test_usecols_implicit_index_col� s r> c C sJ | }d}|j t|�ddd�}tddgddgd�d d gd �}t�||� d S ) Nz-a b c 4 apple bat 5.7 8 orange cow 10z\s+)r( r )�sepr r: r; r< r= r* r )r5 )r r r r r )r r r r r! r r r �test_usecols_regex_sep� s r@ c C sJ | }d}|j t|�ddd�}tddgddgd�d d gd �}t�||� d S ) Nz-a b c 4 apple bat 5.7 8 orange cow 10T)r( r )Zdelim_whitespacer r: r; r< r= r* r )r5 )r r r r r )r r r r r! r r r � test_usecols_with_whitespace� s rA zusecols,expectedi� i� i� i� �2�0)r r �1i� ip c C s* | }d}|j t|�|d�}t�||� d S )Nz#2,0,1 1000,2000,3000 4000,5000,6000)r )r r r r )r r r! r r r r r r �%test_usecols_with_integer_like_header� s rE c C s2 d}t � }| }|jt|�t� d�}t�||� d S )Nza,b,c 1,2,3 4,5,6)r )r r r �setr r )r r r! r r r r r �test_empty_usecols� s rG c C sJ | }d}t �ddg�}tddgg|d�}|jt|�|d�}t�||� d S )Nz a,b,c 1,2,3r( r r r )r )r )�np�arrayr r r r r )r r r r r! r r r r �test_np_array_usecols� s rJ c C s | � � dkS )N)ZAAAZBBBZDDD)�upper)�xr r r �<lambda>� � rM g ����?g��$���@gr.�-� @)r r r r r+ )ZAaAZbBbZdddc C s dS )NFr )rL r r r rM � rN c C s* d}| }|j t|�|d�}t�||� d S )NzPAaA,bBb,CCC,ddd 0.056674973,8,True,a 2.613230982,2,False,b 3.568935038,7,False,a)r )r r r r )r r r! r r r r r r �test_callable_usecols� s rO c C s | dkS )N)r( r r )rL r r r rM rN c C sN d}| }dddg}t ddgtjdgd��}|jt|�||d�}t�||� d S ) Nz 1,2 1,2,3r( r r r r )r( r )r% r )r rH �nanr r r r )r r r r r% r! r r r r �test_incomplete_first_row s rQ z data,usecols,kwargs,expectedz 19,29,39 19,29,39 10,20,30,40r& � � �' r, � � z4A,B,C 1,2,3 3,4,5 1,2,4,5,1,6 1,2,3,,,1, 1,2,3 5,6,7�A�B�Cr r r* r )rW rX rY c C s. | }|j t|�fd|i|��}t�||� d S )Nr )r r r r )r r r �kwargsr! r r r r r �test_uneven_length_cols s r[ zusecols,kwargs,expected,msgr1 )r( r r r1 �fz\['f'\]�gz\[('f', 'g'|'g', 'f')\]�D)r& r% )rW rX rY r^ r% c C sl d}|j |d� | }|d krJtjt|d�� |jt|�f|� W d Q R X n |jt|�f|�}t�||� d S )Nza,b,c,d 1,2,3,4 5,6,7,8)r )r )�updater r r r r r r )r r rZ r! r. r r r r r r �%test_raises_on_usecols_names_mismatch7 s , r` c C sP d}ddddg}| }|j t|�d||d�}tdd gd d gd ��}t�||� d S ) Nza,b,c,d 1,2,3,4 5,6,7,8rW rX rY r^ r )r& r% r r r r r+ )rW rY )r r r r r )r r r r% r r r! r r r �/test_usecols_subset_names_mismatch_orig_columnso s ra c C s� | }d}t jtddd�� |jt|�ddg|dd�}W d Q R X tdgd gd��}|d krp|jd krptd dgi�}t �||� d S ) Nz a,b 1,2 F)Zcheck_stacklevelZraise_on_extra_warningsr r )r r% r& r )r( r �pythonr( )r Zassert_produces_warning� FutureWarningr r r Zenginer )r r% r r r r! r r r �"test_usecols_indices_out_of_boundsz s $rd )'�__doc__�ior �numpyrH r Zpandasr r Zpandas._testingZ_testingr r Z_msg_validate_usecols_namesr �markZ parametrizer" r' r) r- r/ r0 r3 r6 r9 r> r@ rA rE rG rJ rO rQ r[ �formatr` ra rd r r r r �<module> s� . $ "* " $ "
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка