Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/string_/__pycache__/test_string.cpython-37.pyc
Назад
B �5�gfH � @ s� d Z ddlZddlZddlm mZ ddlm Z ddl Z ddl m Z ddlmZ ejdd� �Zejdd� �Zd d � Zd d � Zd d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z ej j dd �d d � �Z ej j dd �d d � �Z d!d"� Z!d#d$� Z"d%d&� Z#d'd(� Z$d)d*� Z%ej �&d+d,d-g�d.d/� �Z'd0d1� Z(d2d3� Z)d4d5� Z*ej �&d6d,d-g�ej j d7d �d8d9� ��Z+ej �&d:d;d<g�ej �&d6d,d-g�d=d>� ��Z,ej �&d:d;d<g�ej �&d?e j-e j.g�d@dA� ��Z/ej �&d6d,d-g�ej j d7d �dBdC� ��Z0dDdE� Z1e�2dF�dGdH� �Z3e�2dF�dIdJ� �Z4e�2dF�dKdL� �Z5dMdN� Z6dOdP� Z7ej �&dQdRdSdTge�.d-d-d-g�fdRdSdge�.d-d-d,g�fg�dUdV� �Z8dWdX� Z9ej �&dYej:ej;ej<g�dZd[� �Z=d\d]� Z>d^d_� Z?d`da� Z@dS )bz� This module tests the functionality of StringArray and ArrowStringArray. Tests for the str accessors are in pandas/tests/strings/test_string_array.py � N)�is_dtype_equal)�ArrowStringArrayc C s t j| d�S )N)�storage)�pd� StringDtype)�string_storage� r �]/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/string_/test_string.py�dtype s r c C s | � � S )N)Zconstruct_array_type)r r r r �cls s r c C s� t �dt jdt jdg| d�i�}d}t|�|ks4t�d}t|j�|ksJt�| jdkrXdnd }d |� d �}t|jj�|ks|t�d S ) N�A�a�b)r z A 0 a 1 <NA> 2 bz40 a 1 <NA> 2 b Name: A, dtype: string�pyarrowr � StringArray�<z+> ['a', <NA>, 'b'] Length: 3, dtype: string)r � DataFrame�array�NA�repr�AssertionErrorr r )r �df�expectedZarr_namer r r � test_repr s r c C s6 | � dd dg�}|d d k s t�|d tjks2t�d S )Nr r � )�_from_sequencer r r )r r r r r �test_none_to_nan( s r c C s� | � ddg�}| tjjkr d}nd}tjt|d�� d|d< W d Q R X | tjjkrXd}nd}tjt|d�� t�d d g�|d d �< W d Q R X d S ) Nr r z4Cannot set non-string value '10' into a StringArray.zScalar must be NA or str)�match� r zMust provide strings.r � ) r r �arraysr �pytest�raises� ValueError�npr )r �arr�msgr r r �test_setitem_validates. s r' c C s< t jddg| d�}d|d< t jddg| d�}t�||� d S )Nr �c)r �dr )r r �tm� assert_extension_array_equal)r r% r r r r � test_setitem_with_scalar_string@ s r, c C s� | j dkr,d}tjj|td�}|j�|� n tjjdtd�}|j�|� t�tj ddd��}d |d< |� | �}t |j | �s~t �|� d �}t�||� d S ) Nr z6ValueError: Could not convert object to NumPy datetime)�reasonr" z/GH#36153 casting from StringArray to dt64 fails�2000� )Zperiodsr zdatetime64[ns])r r! �mark�xfailr# �node� add_markerr �SeriesZ date_range�astyper r r r* �assert_series_equal)r �requestr- r0 �serZcasted�resultr r r �test_astype_roundtripI s r: c C s | j dkr*d}tjjt|d�}|j�|� tjdddd d g| d�}tjdd d d d g| d�}|| }tjd d d d d g| d�}t � ||� |� |�}t � ||� |� |�}tjd dd d d g| d�}t � ||� |j |dd�}tjd d ddd g| d�}t � ||� d S )Nr zLunsupported operand type(s) for +: 'ArrowStringArray' and 'ArrowStringArray')r" r- r r r( )r �x�y�z�axZby�xaZyb�-)Z fill_valuezc-z-z) r r! r0 r1 � TypeErrorr2 r3 r r4 r* r6 �addZradd)r r7 r- r0 r r r9 r r r r �test_add^ s" rC c C s� | j dkr*d}tjjd |d�}|j�|� tjdddg| d�}tjdddggt d�}tj t dd �� || W d Q R X t� |�}tj t dd �� || W d Q R X d S ) Nr z*Failed: DID NOT RAISE <class 'ValueError'>)r" r- r r r( )r z3 != 1)r ) r r! r0 r1 r2 r3 r r r$ �objectr" r# r4 )r r7 r- r0 r r �sr r r � test_add_2dz s rF c C s� | j dkr*d}tjjt|d�}|j�|� tjddd d g| d�}dd dd g}|| }tjd d d d g| d�}t � ||� || }tjd d d d g| d�}t � ||� d S ) Nr z@unsupported operand type(s) for +: 'ArrowStringArray' and 'list')r" r- r r )r r; r< r> r? ) r r! r0 r1 rA r2 r3 r r r* r+ )r r7 r- r0 r �otherr9 r r r r �test_add_sequence� s rH c C s~ | j dkr*d}tjjt|d�}|j�|� tjddd g| d�}|d }tjdd d g| d�}t � ||� d| }t � ||� d S ) Nr z?unsupported operand type(s) for *: 'ArrowStringArray' and 'int')r" r- r r )r r ZaaZbb) r r! r0 r1 rA r2 r3 r r r* r+ )r r7 r- r0 r r9 r r r r �test_mul� s rI zGH-28527)r- c C s� t jddddg| d�}t �dddd gg�}|�|�tks<t�|| }t �d d d d gg��| �}t�||� || }t �ddddgg��| �}t�||� d S )Nr r r( r) )r �t�u�v�w�atZbuZcvZdw�taZub�vc�wd) r r r �__add__�NotImplementedr r5 r* �assert_frame_equal)r r% r r9 r r r r �test_add_strings� s rU c C s� t jddtjtjg| d�}t �dtjdtjgg�}|�|�tksDt�|| }t �dtjtjtjgg��| �}t � ||� || }t �dtjtjtjgg��| �}t � ||� d S )Nr r )r r; r< r> r? ) r r r$ �nanr rR rS r r5 r* rT )r r% r r9 r r r r �test_add_frame� s rW c sf | � t jdd dg|d�}d�t|� ���}tj� �fdd�|D �td�}t j|dd�}t�||� d S )Nr r( )r c s g | ]}t |� ����qS r )�getattr)�.0�item)�op_namerG r r � <listcomp>� s z2test_comparison_methods_scalar.<locals>.<listcomp>�boolean)r r rX r$ rD r* r+ )�all_compare_operatorsr r r9 r r )r[ rG r � test_comparison_methods_scalar� s r_ c C sL | }t jdd dg|d�}t||�t j�}t jd d d gdd�}t�||� d S )Nr r( )r r] )r r rX r r* r+ )r^ r r[ r r9 r r r r �$test_comparison_methods_scalar_pd_na� s r` c C s� | dkr(d}t jjt|d�}|j�|� | }tjdd dg|d�}d}t||�|�}dd dgd d d gd�| } tj| d d�} t � || � d S ) N)�__eq__�__ne__z@comparison op not supported between instances of 'str' and 'int')r" r- r r( )r �* FTr] ) r! r0 r1 rA r2 r3 r r rX r* r+ ) r^ r r7 r- r0 r[ r rG r9 Z expected_datar r r r �)test_comparison_methods_scalar_not_string� s rd c C s� |j dkr&tjjtdd�}|j�|� | }tjdd dg|d�}d d dg}t ||�|�}t j |dd�}t |d |�|d �|d< tj|d d�}t � ||� t ||�tj�}tjd d d gd d�}t � ||� d S ) Nr z left is not an ExtensionArray)r" r- r r( )r rD ���r] )r r! r0 r1 r r2 r3 r r rX r$ Z empty_liker* r+ r ) r^ r r7 r0 r[ r rG r9 r r r r � test_comparison_methods_array� s rf c C s | t jjkrd}nd}tjt|d�� | tjddgdd�� W d Q R X tjt|d�� | t�g �� W d Q R X tjt|d�� | tjdtjgt d�� W d Q R X tjt|d�� | tjdd gt d�� W d Q R X tjt|d�� | tjdt j gt d�� W d Q R X d S )Nz7StringArray requires a sequence of strings or pandas.NAz?Unsupported type '<class 'numpy.ndarray'>' for ArrowStringArray)r r r ZS1)r ) r r r r! r" r# r$ r rV rD ZNaT)r r& r r r �test_constructor_raises s " rg �copyTFc C s� |t kr,| dkr,tjjtdd�}|j�|� tjdtj gt d�}tjdt j gt d�}|j || d�}|t kr�dd l}||j||�� dd ��}n||�}t�||� | r�|n|}t�||� d S ) NFznumpy array are different)r" r- r )r )rh r T)�type� from_pandas)r r! r0 r1 r r2 r3 r$ r rV rD r r r r �stringr* r+ �assert_numpy_array_equal) rh r r7 r0 Znan_arrZna_arrr9 �par r r r � test_from_sequence_no_mutate s rn c C s� t jdddg| d�}|�d�}tjdddgdd�}t�||� t jdt jdg| d�}d }tjt |d �� |�d� W d Q R X d S ) N�1�2�3)r Zint64r r � zJint\(\) argument must be a string, a bytes-like object or a( real)? number)r ) r r r5 r$ r* rl r r! r" rA )r r% r9 r r& r r r �test_astype_int3 s rs c C sF t jdt jdg| d�}|�d�}t jdt jdgdd�}t�||� d S )Nro rq )r �Int64r rr )r r r r5 r* r+ )r r% r9 r r r r �test_astype_nullable_int? s ru c C sF t jdt jdg| d�}|�|�}t jdtjdg|d�}t�||� d S )Nz1.1z3.3)r g�������?gffffff @)r r4 r r5 r$ rV r* r6 )r Z any_float_allowed_nullable_dtyper8 r9 r r r r �test_astype_floatG s rv �skipnaz Not implemented StringArray.sumc C s0 t jdddg|d�}|j| d�}|dks,t�d S )Nr r r( )r )rw �abc)r r4 �sumr )rw r r% r9 r r r � test_reduceO s rz �method�min�maxc C s� |j dkr*d}tjjt|d�}|j�|� tjdddd g|d�}t || �|d�}|rr| d kr`dnd}||ks�t �n|tj ks�t �d S ) Nr z0'ArrowStringArray' object has no attribute 'max')r" r- r r r( )r )rw r| ) r r! r0 r1 �AttributeErrorr2 r3 r r4 rX r r ) r{ rw r r7 r- r0 r% r9 r r r r � test_min_maxW s r �boxc C s� |j dkrB|tjkr t}d}nt}d}tjj||d�}|j� |� |dddd g|d�}t t | �|�}| d krpdnd} || ks�t �d S ) Nr z<'<=' not supported between instances of 'str' and 'NoneType'z0'ArrowStringArray' object has no attribute 'max')r" r- r r r( )r r| ) r r r rA r~ r! r0 r1 r2 r3 rX r$ r ) r{ r� r r7 r" r- r0 r% r9 r r r r �test_min_max_numpyh s r� c C sJ t jd dd ddd g|d�}|j| d�}| r8|dksFt�nt �|�sFt�d S )Nr r r( )r )rw rx )r r4 ry r �isna)rw r r% r9 r r r �test_reduce_missing{ s r� c C s� | j dkr*d}tjjt|d�}|j�|� tjdtj g| d�}|j dd�}tjddg| d�}t � ||� |j t �d�d�}tjddg| d�}t � ||� d}tjt|d �� |j d d� W d Q R X d S ) Nr zmRegex pattern "Cannot set non-string value '1' into a StringArray." does not match 'Scalar must be NA or str')r" r- r )r r )�valuez3Cannot set non-string value '1' into a StringArray.)r r )r r! r0 r1 r r2 r3 r r r Zfillnar* r+ r$ Zstr_r" r# )r r7 r- r0 r% �resr r& r r r �test_fillna_args� s r� r c C sd dd l }tjdddg| d�}|�|�}|jt|�|�� dd�}| jdkrR|�|�}|�|�s`t�d S ) Nr r r r( )r T)ri rj r ) r r r �listrk r � chunked_array�equalsr )r rm �datar% r r r r �test_arrow_array� s r� c C s� dd l }tjddd g| d�}t�d|i�}|�|�}|�d�jdksHt�t�d|�� |� � }W d Q R X t |d j tj �s~t�|� d|� d��}t�||� |jd tjks�t�d S ) Nr r r )r rk r zstring[�])r r )r r r r �table�fieldri r �option_context� to_pandas� isinstancer r r5 r* rT �locr )r �string_storage2rm r� r r� r9 r r r r �test_arrow_roundtrip� s r� c C s� dd l }tjg | d�}t�d|i�}|�|�}|�d�jdksBt�|j|jg |� � d�g|j d�}t� d|�� |� � }W d Q R X t |d jtj�s�t�|�d|� d ��}t�||� d S ) Nr )r r rk )ri )�schemar zstring[r� )r r r r r� r� ri r r� rk r� r� r� r� r r r5 r* rT )r r� rm r� r r� r9 r r r r � test_arrow_load_from_zero_chunks� s r� c C s� t jdddt jg| d�}|jdd�}t jdddgddt jgdd �}t�||� |jd d�}t jddgddgdd �}t�||� d S ) Nr r )r F)Zdropnar r rt )�indexr T)r r r � value_countsr4 r* r6 )r r% r9 r r r r �test_value_counts_na� s r� c C sP t jdddt jg| d�}|jdd�}t jddgddgdd �d }t�||� d S ) Nr r )r T)� normalizer r ZFloat64)r� r rr )r r4 r r� r* r6 )r rE r9 r r r r � test_value_counts_with_normalize� s r� zvalues, expectedr r r( c C s� t j| |d�} t �dd��b | �� }t�||� t �| ��� }t �|�}t�||� t �| ��� }t �|�}t� ||� W d Q R X d S )N)r zmode.use_inf_as_naT) r r r� r� r* rl r4 r6 r rT )�valuesr r r9 r r r �test_use_inf_as_na� s r� c C s^ | j dkrt�d� tjdddg| d�}d|j k rT|�� krT|jdd �k sZn t�d S ) Nr znot applicabler r r( )r r T)�deep)r r! �skipr r4 �nbytesZ memory_usager )r Zseriesr r r �test_memory_usage� s r� � float_dtypec C s: t jdg| d�}|�|�}t jdg|d�}t�||� d S )Ng�������?)r z0.1)r r4 r5 r* r6 )r� r rE r9 r r r r � test_astype_from_float_dtype s r� c C sF t jdt jdg| d�}t�|�}tjdt jdgtd�}t�||� d S )Nr r )r )r r r r$ rD r* rl )r r% r9 r r r r �"test_to_numpy_returns_pdna_default s r� c C sJ |}t jdt jdg| d�}|j|d�}tjd|dgtd�}t�||� d S )Nr r )r )�na_value)r r r Zto_numpyr$ rD r* rl )r Z nulls_fixturer� r% r9 r r r r �test_to_numpy_na_value s r� c C s� t jddd g| d�}|�ddg�}t �dddg�}t�||� |�dt jg�}t �dddg�}t�||� |�g �}t �dddg�}t�||� |�dt j�� g�}t �dddg�}t�||� d S )Nr r )r r( TF)r r4 �isinr* r6 r � Timestamp�now)r r7 rE r9 r r r r � test_isin s r� )A�__doc__�numpyr$ r! Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandas.core.dtypes.commonr Zpandasr Zpandas._testingZ_testingr* Z pandas.core.arrays.string_arrowr Zfixturer r r r r' r, r: rC rF rH rI r0 r1 rU rW r_ r` rd rf rg Z parametrizern rs ru rv rz r r4 r r� r� r� Z skip_if_nor� r� r� r� r� r� r� Zfloat16Zfloat32Zfloat64r� r� r� r� r r r r �<module> sh & "
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка