Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/arrays/masked/__pycache__/test_arrow_compat.cpython-37.pyc
Назад
B �5�g� � @ s, d dl Zd dlZd dlm mZ d dlZd dl m Z ej ddd�Z d dlmZ dd� e jD �Zedd� e jD �7 Zeejd d d dgd d �g7 Zejed d� eD �d�dd� �Zdd� Ze�d�dd� �Ze�d�dd� �Ze�d�dd� �Ze�d�dd� �Zejdd � �Z d d � Z e�d�d d � �Z dS )!� NZpyarrowz0.17.0)Z minversion)� pyarrow_array_to_numpy_and_maskc C s" g | ]}t jd dddg|d��qS )� � � N)�dtype)�pd�array)�.0r � r �b/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/arrays/masked/test_arrow_compat.py� <listcomp> s r c C s" g | ]}t jd dddg|d��qS )g�������?g�������?g333333�?N)r )r r )r r r r r r s TF�boolean)r c C s g | ] }|j j�qS r )r �name)r �ar r r r s )�paramsZidsc C s | j S )N)�param)�requestr r r �data s r c C s> t �| �}t j| jtd d�t �| jj�d�}|�|�s:t�d S )N)Zna_value)�type) �par Zto_numpy�object�from_numpy_dtyper � numpy_dtype�equals�AssertionError)r �arr�expectedr r r �test_arrow_array s r c C s` t �d| i�}t�|�}|�d�jt| jj�ks4t �|� � }|d j| jksPt �t � ||� d S )Nr ) r � DataFramer �table�fieldr �strr r r � to_pandas�tm�assert_frame_equal)r �dfr �resultr r r �test_arrow_roundtrip s r' c C s� t �d| dd� i�}t�|�}|�d�jt| jj�ks<t �tjtj g |�d�jd�g|j d�}|� � }|d j| jks|t �t �||� d S )Nr r )r )�schema)r r r r r r r! r r r Z chunked_arrayr( r" r# r$ )r r% r r& r r r � test_arrow_load_from_zero_chunks* s r) c C sN t �� } | �tjddddd gdd��}t jddddd gdd�}t�||� d S ) Nr r r � Zint64)r ZUInt32)r )r Z UInt32Dtype�__from_arrow__r r r# Z assert_extension_array_equal)r r& r r r r �test_arrow_from_arrow_uint9 s r, c C s� t �d| i�}t�|�}|�dd ��� }|jdd � jdd�}t� ||� |� | d �}t�|�}|�dd ��� }|jdd � jdd�}t� ||� d S )Nr r T)Zdropr ) r r r r �slicer" ZilocZ reset_indexr# r$ Zfillna)r r% r r& r Zdf2r r r �test_arrow_slicedE s r. c C s\ t �| �}t�|�}tjdddd g|d�}t jdddg|d�}t �ddddg�}||||fS )Nr r r )r )r TF)�npr r r r )Zany_real_dtype�np_dtypeZpa_type�pa_array� np_expected� mask_expectedr r r �np_dtype_to_arraysW s r4 c C s� | \}}}}t ||�\}}t�|dd� |� t�||� |�� d }|�� d }|�� d �� } t�| d �} tjj|j t |�|| g|j d�} | � � t | |�\}}t�|dd� |� t�||� d|j j d } t�| | �} t�d�}tjj|j t |�|| g|j d d�}|� � t ||�\}}t�|dd� |� t�||� tjg |d �}tjg tjd �}tjj|j d||g|j d�}|� � t ||�\}}t�|dd� |� t�||� dS ) a& Test conversion from pyarrow array to numpy array. Modifies the pyarrow buffer to contain padding and offset, which are considered valid buffers by pyarrow. Also tests empty pyarrow arrays with non empty buffers. See https://github.com/pandas-dev/pandas/issues/40896 Nr r r � )r �length�buffers�offset� � )r )r r# Zassert_numpy_array_equalr7 Z to_pybytesr Z py_bufferZArrayZ from_buffersr �lenr8 �validateZ bit_widthr/ r Zbool_)r4 r0 r1 r2 r3 r �maskZ mask_bufferZ data_bufferZdata_buffer_bytesZdata_buffer_trailZpa_array_trailr8 Zdata_buffer_offsetZmask_buffer_offsetZpa_array_offsetZnp_expected_emptyZmask_expected_emptyr r r �$test_pyarrow_array_to_numpy_and_maske sP r> c C s\ |j dkr"| j�tjjd dd�� t�|��d�}tj t d d�� |j � |� W d Q R X d S )Nr z#numeric dtypes don't error but cast)�raises�reason�string)�match) r �nodeZ add_marker�pytest�markZxfailr r �castr? � TypeErrorr+ )r r r r r r �test_from_arrow_type_error� s rH ) �numpyr/ rD Z pandas.util._test_decorators�utilZ_test_decorators�tdZpandasr Zpandas._testingZ_testingr# Z importorskipr Z pandas.core.arrays._arrow_utilsr ZALL_EA_INT_DTYPESZarraysZFLOAT_EA_DTYPESr Zfixturer r Z skip_if_nor' r) r, r. r4 r> rH r r r r �<module> s$ ?
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.05 |
proxy
|
phpinfo
|
Настройка