__init__.pyc
1 o 2 c��c5� � @ sX U d dl mZ d dlZd dlmZ d dlmZ d dlZd dlZd dlZd dl Z d dl 3 m Z d dl m Z mZmZmZmZ d dlZd dlZd dlmZmZmZ d dlmZ d d lm Z d d 4 l m Z m Z m Z m!Z!m"Z" d dl#Z$d d l#m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z. d d l/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5 d d l6m7Z7m8Z8m9Z9 d dl:m;Z;m<Z< d dl=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZT d dlUmVZVmWZW d dlXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z` d dlambZbmcZcmdZdmeZe d dlfmgZgmhZhmiZi d dljmkZk d dllmmZm e �r d dl#mnZnmoZo dZpdZqg d�Zrdesd< g d �Ztdesd < eud d d d!gZvdesd"< g d#�Zwdesd$< erev Zxetew Zyezd%d&gZ{desd'< d(d)gZ|desd*< e}d+d,gZ~desd-< ed.d/gZ�desd0< d1d2gZ�desd3< d4d5gZ�desd6< e�d7gZ�desd8< e�d9gZ�desd:< e�d;gZ�desd<< e{ex Z�e�e~ e� e� e� e� e� e� Z�ej�ej�ej�ej�ej�ej�ej�ej�gZ�d=d>d?�e Z�dej�e$j�ezd@�e$j�edA�gZ�dBdC� ej�ej�fD �Z�e �sSd dl�Z�e���� e���� e���� e���� gZ�e���� e���� e���� e���� gZ�e�e� Z�e���� e���� gZ�e�� � e���� gZ�e���dD�e���dE�e���dF�e���dG�gZ�e���� e���� gZ�dHdC� dID �Z�dJdC� dID �Z�e���� gZ�e�e� e� e� e� e� e� Z�e��dK�Z�e�e�fZ�d�dNdO�Z�d�dPdQ�Z�e�� d�dRdS�Z�d�dTdU�Z�d�dWdX�Z�dYdZ� Z�d�d[d\�Z�d�d�d_d`�Z�d�d�dcdd�Z�d�d�dfdg�Z�d�d�dhdi�Z�d�d�dkdl�Z�d�d�dndo�Z�d�d�dqdr�Z�d�d�dtdu�Z�d�d�dwdx�Z�d�d�d}d~�Z�d�d�d�d��Z�d�d�d�d��Z�d�d�d��Z�d�d�� Z�d�d�d�d��Z�dej�fd�d�d��Z�d�d�d�d��Z�d�d�d�d��Z�d�d�d�d��Z�d�d�d��Z�d�d�d�d��Z�d�d�d�d��Z�d�d�d�d��Z�d�d�d�d��Z�d�d�d�d��Z�d�d�d��Z�d�d�� Z�d�d�d��Z�d�d�d�d��Z� � � d�d�d�d��Z� V V � � d�d�d�d��Z�d�d�d��Z�d�d�d�d��Z�G d�d�� d�e-�Z�G d�d�� d�e'�Z�G d�d�� d�e%�Z�d�d�d��Z�d d�dńZ�dd�dʄZ�e$j�j�j�� Z�d�d̄ Z�dd�dЄZ�d�d҄ Z�d�dԄ Z�d�dք Z�d�d Z�d�dڄ Z�d�d܄ Z�d�d�dބZ�g dߢZ�dS ( � )� annotationsN)�datetime)�Decimal)� byteorder)� TYPE_CHECKING�Callable�ContextManager�Counter�Iterable)�can_set_locale� get_locales� 5 set_locale)�Dtype)�pa_version_under1p01)�is_float_dtype�is_integer_dtype� is_sequence�is_unsigned_integer_dtype� pandas_dtype) 6 � Categorical�CategoricalIndex� DataFrame� DatetimeIndex�Index� IntervalIndex� 7 MultiIndex� 8 RangeIndex�Series� bdate_range)�close�network�round_trip_localpath�round_trip_pathlib�round_trip_pickle�write_to_compressed)�randbool�rands� rands_array)�assert_produces_warning�maybe_produces_warning)�assert_almost_equal�assert_attr_equal�assert_categorical_equal�assert_class_equal�assert_contains_all� assert_copy�assert_datetime_array_equal�assert_dict_equal� assert_equal� assert_extension_array_equal�assert_frame_equal�assert_index_equal�!assert_indexing_slices_equivalent�assert_interval_array_equal�assert_is_sorted�"assert_is_valid_plot_return_object�assert_metadata_equivalent�assert_numpy_array_equal�assert_period_array_equal�assert_series_equal�assert_sp_array_equal� assert_timedelta_array_equal�raise_assert_detail)� get_dtype�get_obj)� 9 RNGContext�decompress_file� ensure_clean�ensure_clean_dir�!ensure_safe_environment_variables� set_timezone� use_numexpr�with_csv_dialect)� Float64Index� 10 Int64Index� NumericIndex� UInt64Index)�BaseMaskedArray�ExtensionArray� PandasArray)�NDArrayBackedExtensionArray)� extract_array)� PeriodIndex�TimedeltaIndex� � )�uint8�uint16�uint32�uint64z list[Dtype]�UNSIGNED_INT_NUMPY_DTYPES)�UInt8�UInt16�UInt32�UInt64�UNSIGNED_INT_EA_DTYPES�int8�int16�int32�int64�SIGNED_INT_NUMPY_DTYPES)�Int8�Int16�Int32�Int64�SIGNED_INT_EA_DTYPES�float32�float64�FLOAT_NUMPY_DTYPES�Float32�Float64�FLOAT_EA_DTYPES� complex64� 11 complex128�COMPLEX_DTYPES�str�U� STRING_DTYPESzdatetime64[ns]zM8[ns]�DATETIME64_DTYPESztimedelta64[ns]zm8[ns]�TIMEDELTA64_DTYPES�bool� BOOL_DTYPES�bytes� BYTES_DTYPES�object� OBJECT_DTYPES�<�>)�little�big�nan�NaNc C s g | ] }d D ]}|d|��qqS )) �Y�M�W�D�h�m�s�ms�us�ns�ps�fs�as�NaT� )�.0�cls�unitr� r� ��C:\Users\Jacks.GUTTSPC\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\pandas\_testing\__init__.py� 12 <listcomp>� s ���r� r� r� r� r� c C s$ g | ]}d D ] }t j||d��qqS ))N�UTCz 13 US/Pacificz 14 US/Eastern)r� �tz)�pa� timestamp)r� r� r� r� r� r� r� � s � ��)r� r� r� r� c C s g | ]}t �|��qS r� )r� �duration)r� r� r� r� r� r� � � z^$�return�Nonec C �4 t j�dd�} d| v rtD ] 15 }t�d|� q d S d S )N�PANDAS_TESTING_MODEr� � deprecate�always��os�environ�get�_testing_mode_warnings�warnings� simplefilter�Z testing_mode�categoryr� r� r� �set_testing_mode� � �r� c C r� )Nr� r� r� �ignorer� r� r� r� r� �reset_testing_mode� r� r� c C s t jddd� dS )zJ 16 Reset the display options for printing and representing objects. 17 z ^display.T)�silentN)�pd� reset_optionr� r� r� r� �reset_display_options s r� c C s t | �t |�kS )zO 18 Checks if the set of unique elements of arr1 and arr2 are equivalent. 19 )� frozenset)�arr1�arr2r� r� r� � equalContents s r� Tc C s� |t ju rt| t�rtt�| j��} | S t �| �} | S |tu r&t� | �} | S |t 20 u r0t 21 | �} | S |t u rKt 22 | �� � } |rI| j } t j| gd dd�} | S |tju sU|tju r\t�| �} | S |tu rft| �} | S t|��)z� 23 Helper function to wrap the expected output of a test in a given box_class. 24 25 Parameters 26 ---------- 27 expected : np.ndarray, Index, Series 28 box_cls : {Index, Series, DataFrame} 29 30 Returns 31 ------- 32 subclass of box_cls 33 � T)� ignore_index)r� �array� 34 isinstancer rQ �np�asarray�_valuesr � _with_inferr r �to_frame�T�concat�ndarray�to_array�NotImplementedError)�expectedZbox_cls� transposer� r� r� � box_expected s2 35 36 37 �� 38 �� � 39 ��r� c C s* t | dd�}|du rt�| �S t| dd�S )zQ 40 Similar to pd.array, but does not cast numpy dtypes to nullable dtypes. 41 �dtypeNT)� extract_numpy)�getattrr� r� rS )�objr� r� r� r� r� ? s 42 r� c C s t jd | � S �N)�string�ascii_uppercase��kr� r� r� �getColsP � r� � 43 r c C s t td| d�|d�S )Nr� )�nchars�size��name)r r' �r� r� r� r� r� �makeStringIndexU s r� � r c K s6 t d|dd�}ttjt�| �| |d�fd|i|��S )z%make a length k index or n categoriesrW F)r� r� �replace)� 44 categoriesr� )r' r r � 45 from_codesr� �arange)r� �nr� �kwargs�xr� r� r� �makeCategoricalIndexY s ���r� r c K s* t jdd| d d�}tj|fd|i|��S )z make a length k IntervalIndexr �d � )�numr� )r� �linspacer � from_breaks)r� r� r� r� r� r� r� �makeIntervalIndexa s r� c C sL | dkr t dg|d�S | dkrt ddg|d�S t ddgdg| d |d�S )Nr� Tr� r� F)r r� r� r� r� � makeBoolIndexg s 46 r� rM c C s� t |�}t|tj�s J �t|�r'tj| |d�}t|�r&|d|jd d 7 }n't|�rGtj � 47 | �tj � 48 d� }|� � |dtj � dd� }nt d|� ���t|||d �S ) 49 N�r� r� � r� r� r � z wrong dtype )r� r� )r r� r� r� r r� r �itemsizer �random� random_sample�sort�randintr� rM )r� r� r� �valuesr� r� r� �makeNumericIndexo s �r� rL c C � t | |dd�}t|�S )Nre �r� r� )r� rL �r� r� Zbase_idxr� r� r� � makeIntIndex� � r� rN c C r� )Nr[ r� )r� rN r� r� r� r� � makeUIntIndex� r� r� r c K s t d| dfd|i|��S )Nr r� r� )r )r� r� r� r� r� r� �makeRangeIndex� � r� rK c C r� )Nrm r� )r� rK r� r� r� r� �makeFloatIndex� r� r �Br� �intr c K s0 t ddd�}t|| ||d�}t|fd|i|��S )N�� r� )�periods�freqr� r� )r r r )r� r r� r� �dt�drr� r� r� � makeDateIndex� s r r� rU c K s t jdd| ||d�|��S )Nz1 day��startr r r� r� )r� �timedelta_range)r� r r� r� r� r� r� �makeTimedeltaIndex� s r rT c K s& t ddd�}tjd|| d|d�|��S )Nr r� r r r� )r r� � period_range)r� r� r� r r� r� r� �makePeriodIndex� s r c K sJ | d d }t |�}tjd|gfd|i|��}t|�| ks J �|d | � S )Nr� r� )�foo�bar�names)�ranger � from_product�len)r� r r� �N�rng�mir� r� r� �makeMultiIndex� s 50 r c c s" � t ttttttg} | E d H d S r� )r r r r� r� r� r )�make_index_funcsr� r� r� � index_subclass_makers_generator� s �� r �Iterable[Index]c c s& � t ttg}|D ]}|| d�V qdS )z� 51 Generator which can be iterated over to get instances of all the classes 52 which represent time-series. 53 54 Parameters 55 ---------- 56 k: length of each of the index instances 57 r� N)r r r )r� r Zmake_index_funcr� r� r� � all_timeseries_index_generator� s � 58 ��r r c C s\ t t�}tj�t�}tjdd�� |j|dd�}W d � n1 s"w Y t||| d�S )Nr� )�invalidF)�copy��indexr� )r� �_Nr� r� �randn�errstate�astyper )r� r� r �datar� r� r� �make_rand_series� s �r& c C � 59 t | d�S �Nr� �r& r� r� r� r� �makeFloatSeries� � 60 r* c C r' r( r) r� r� r� r� �makeStringSeries� r+ r, c C s* t t�}t|td�}t t�}t||| d�S )Nr� r )r� r! r r~ r )r� r% r r� r� r� �makeObjectSeries� s r- �dict[str, Series]c s t t�� � fdd�tt�D �S )Nc s" i | ] }|t tj�t�� d ��qS )�r )r r� r� r" r! �r� �cr/ r� r� � 61 <dictcomp>� s " z!getSeriesData.<locals>.<dictcomp>)r� r! r� �_Kr� r� r/ r� � getSeriesData� s r4 c C s* | d u rt } ttj�| �t| |d�|d�S )N)r r )r! r r� r� r" r )�nperr r� r� r� r� �makeTimeSeries� s 62 �r6 c C s& | d u rt } ttj�| �t| �|d�S )Nr )r! r r� r� r" r )r5 r� r� r� r� �makePeriodSeries� s r7 c s � �fdd�t t�D �S )Nc s i | ]}|t �� ��qS r� )r6 r0 �r r5 r� r� r2 � � z%getTimeSeriesData.<locals>.<dictcomp>�r� r3 )r5 r r� r8 r� �getTimeSeriesData� r� r; c s � fdd�t t�D �S )Nc s i | ]}|t � ��qS r� )r7 r0 �r5 r� r� r2 � r� z!getPeriodData.<locals>.<dictcomp>r: r< r� r< r� � getPeriodData� s r= r c C s t | |�}t|�S r� )r; r )r5 r r% r� r� r� �makeTimeDataFrame� s 63 r> c C s t � } t| �S r� )r4 r )r% r� r� r� � makeDataFrame s r? c C s6 t g d��} g d�g d�g d�tddd�d�}| |fS ) N)�a�br1 �d�e)� � �?g @g @g @)rD rE rD rE rD )�foo1�foo2Zfoo3Zfoo4Zfoo5z1/1/2009� )r )�Ar �Cr� )r r )r r% r� r� r� �getMixedTypeDict s 64 �rK c C s t t� d �S )Nr� )r rK r� r� r� r� �makeMixedDataFrame r� rL c C s t | �}t|�S r� )r= r )r5 r% r� r� r� �makePeriodFrame s rM �#Fr � bool | str | list[str] | Nonec sZ |du r dg| }t |�rt|�|ksJ �|du s)|du s)|du s)t|�|u s)J �|du s7|dv r5|dks7J �|du rF� fdd�t|�D �}|du rLd}t|t�rX|dkrX|g}ttttt t 65 d�}|� |�}|ru|| �}|rs|d |_ |S |dur�t t|�� d 66 ���t|�|k r�|�dg|t|� � t|�|ks�J �td d � |D ��s�J �g } t|�D ]:} 67 d d� } | || 68 d } t�� } t| �D ]}� � d| 69 � d|� �}|| 70 | |< q�t| �� | d�d| � }| �|� q�tt| � �}| dk�r|du r�dn|d }t|d |d�}|S |dk�r$|du �rdn|d }tdd � |D �|d�}|S tj||d�}|S )a� 71 Create an index/multindex with given dimensions, levels, names, etc' 72 73 nentries - number of entries in index 74 nlevels - number of levels (> 1 produces multindex) 75 prefix - a string prefix for labels 76 names - (Optional), bool or list of strings. if True will use default 77 names, if false will use no names, if a list is given, the name of 78 each level in the index will be taken from the list. 79 ndupe_l - (Optional), list of ints, the number of rows for which the 80 label will repeated at the corresponding level, you can specify just 81 the first few, the rest will use the default ndupe_l of 1. 82 len(ndupe_l) <= nlevels. 83 idx_type - "i"/"f"/"s"/"dt"/"p"/"td". 84 If idx_type is not None, `idx_nlevels` must be 1. 85 "i"/"f" creates an integer/float index, 86 "s" creates a string 87 "dt" create a datetime index. 88 "td" create a datetime index. 89 90 if unspecified, string labels will be generated. 91 Nr� FT)�i�fr� �ur �p�tdc s g | ]}� t |� �qS r� )ru )r� rP ��prefixr� r� r� F r9 z#makeCustomIndex.<locals>.<listcomp>)rP rQ r� r rT rS r zE is not a legal value for `idx_type`, use 'i'/'f'/'s'/'dt'/'p'/'td'.c s s � | ]}|d kV qdS �r Nr� �r� r� r� r� r� � <genexpr>i � � z"makeCustomIndex.<locals>.<genexpr>c S s* dd l }|�dd| ��d�}dd� |D �S )Nr z[^\d_]_?� �_c S s g | ]}t |��qS r� )r )r� r� r� r� r� r� r s z4makeCustomIndex.<locals>.keyfunc.<locals>.<listcomp>)�re�sub�split)r� r] Z numeric_tupler� r� r� �keyfuncn s z makeCustomIndex.<locals>.keyfunc�_l�_g)�keyr� c s s � | ]}|d V qdS rW r� rX r� r� r� rY � rZ )r )r r r r� ru r� r r� r r r r� r� � 92 ValueError�repr�extend�all� collectionsr �sorted�elements�append�list�zipr r � from_tuples)Znentries�nlevelsrV r �ndupe_l�idx_typeZ idx_func_dictZidx_func�idxZ list_of_listsrP r` Z 93 div_factor�cnt�j�label�result�tuplesr� r r� rU r� �makeCustomIndex sl 94 (� 95 96 � 97 98 ��rx r� c s� |dksJ �|dks J �| du s| dv r|dksJ �| 99 du s(| 100 dv r&|dks(J �t �|d||| 101 d�} t | |d||| d�} � du rDdd � � � �fd 102 d �t| �D �}t|| | | d �S ) a� 103 104 Create a DataFrame using supplied parameters. 105 106 Parameters 107 ---------- 108 nrows, ncols - number of data rows/cols 109 c_idx_names, idx_names - False/True/list of strings, yields No names , 110 default names or uses the provided names for the levels of the 111 corresponding index. You can provide a single string when 112 c_idx_nlevels ==1. 113 c_idx_nlevels - number of levels in columns index. > 1 will yield MultiIndex 114 r_idx_nlevels - number of levels in rows index. > 1 will yield MultiIndex 115 data_gen_f - a function f(row,col) which return the data value 116 at that position, the default generator used yields values of the form 117 "RxCy" based on position. 118 c_ndupe_l, r_ndupe_l - list of integers, determines the number 119 of duplicates for each label at a given level of the corresponding 120 index. The default `None` value produces a multiplicity of 1 across 121 all levels, i.e. a unique index. Will accept a partial list of length 122 N < idx_nlevels, for just the first N levels. If ndupe doesn't divide 123 nrows/ncol, the last label might have lower multiplicity. 124 dtype - passed to the DataFrame constructor as is, in case you wish to 125 have more control in conjunction with a custom `data_gen_f` 126 r_idx_type, c_idx_type - "i"/"f"/"s"/"dt"/"td". 127 If idx_type is not None, `idx_nlevels` must be 1. 128 "i"/"f" creates an integer/float index, 129 "s" creates a string index 130 "dt" create a datetime index. 131 "td" create a timedelta index. 132 133 if unspecified, string labels will be generated. 134 135 Examples 136 -------- 137 # 5 row, 3 columns, default names on both, single index on both axis 138 >> makeCustomDataframe(5,3) 139 140 # make the data a random int between 1 and 100 141 >> mkdf(5,3,data_gen_f=lambda r,c:randint(1,100)) 142 143 # 2-level multiindex on rows with each label duplicated 144 # twice on first level, default names on both axis, single 145 # index on both axis 146 >> a=makeCustomDataframe(5,3,r_idx_nlevels=2,r_ndupe_l=[2]) 147 148 # DatetimeIndex on row, index with unicode labels on columns 149 # no names on either axis 150 >> a=makeCustomDataframe(5,3,c_idx_names=False,r_idx_names=False, 151 r_idx_type="dt",c_idx_type="u") 152 153 # 4-level multindex on rows with names provided, 2-level multindex 154 # on columns with default labels and default names. 155 >> a=makeCustomDataframe(5,3,r_idx_nlevels=4, 156 r_idx_names=["FEE","FIH","FOH","FUM"], 157 c_idx_nlevels=2) 158 159 >> a=mkdf(5,3,r_idx_nlevels=2,c_idx_nlevels=4) 160 r N)rP rQ r� r rS rT r� rJ )ro rV r rp rq �Rc S s d| � d|� �S )Nry rJ r� )�rr1 r� r� r� �<lambda>� s z%makeCustomDataframe.<locals>.<lambda>c s$ g | ]� �� fd d�t ��D ��qS )c s g | ]}� �|��qS r� r� r0 )� 161 data_gen_frz r� r� r� � r� z2makeCustomDataframe.<locals>.<listcomp>.<listcomp>)r )r� �r| �ncols)rz r� r� � s $ z'makeCustomDataframe.<locals>.<listcomp>r� )rx r r )�nrowsr~ Z c_idx_namesZ r_idx_namesZ c_idx_nlevelsZ r_idx_nlevelsr| Z c_ndupe_lZ r_ndupe_lr� Z 162 c_idx_typeZ 163 r_idx_type�columnsr r% r� r} r� �makeCustomDataframe� s4 H �� 164 r� c s� |d u rt j}nt j�|�}td| � � ��d}d}t�| |� �}� ��fdd�}|||�}|j�k rE|d9 }|||�}|j�k s7t �|d � ��t�} || � �t�} 165 | 166 � � | � � fS )Nr� rH gR� ��Q�?c s. | � t|��}t�t�|� � ��d �� S r� )�randr r� �unique�floor)r Z _extra_size�ind�r~ r r� r� r� �_gen_unique_rand s z-_create_missing_idx.<locals>._gen_unique_randg�������?rE ) 167 r� r� � RandomState�round�minr� r� r$ r �tolist) r r~ �density� random_state�min_rows�fac� 168 extra_sizer� r� rt rP r� r� r� �_create_missing_idx� s 169 170 171 172 �r� ��������?c C s0 t � }t|j| |d��\}}tj|j||f< |S )N)r� r� )r? r� �shaper� r� r� )r� r� �dfrP rt r� r� r� �makeMissingDataframe s r� c @ s, e Zd ZddgZedd� �Zedd� �ZdS )�SubclassedSeries�testattrr� c C � dd� S )Nc _ � t | i |��S r� �r� ��argsr� r� r� r� r{ ' � z/SubclassedSeries._constructor.<locals>.<lambda>r� ��selfr� r� r� � _constructor s z SubclassedSeries._constructorc C r� )Nc _ r� r� ��SubclassedDataFramer� r� r� r� r{ + r� z9SubclassedSeries._constructor_expanddim.<locals>.<lambda>r� r� r� r� r� �_constructor_expanddim) � z'SubclassedSeries._constructor_expanddimN)�__name__� 173 __module__� __qualname__� _metadata�propertyr� r� r� r� r� r� r� s 174 r� c @ s* e Zd ZdgZedd� �Zedd� �ZdS )r� r� c C r� )Nc _ r� r� r� r� r� r� r� r{ 3 r� z2SubclassedDataFrame._constructor.<locals>.<lambda>r� r� r� r� r� r� 1 r� z SubclassedDataFrame._constructorc C r� )Nc _ r� r� r� r� r� r� r� r{ 7 r� z9SubclassedDataFrame._constructor_sliced.<locals>.<lambda>r� r� r� r� r� �_constructor_sliced5 r� z'SubclassedDataFrame._constructor_slicedN)r� r� r� r� r� r� r� r� r� r� r� r� . s 175 r� c @ s e Zd Zedd� �ZdS )�SubclassedCategoricalc C s t S r� )r� r� r� r� r� r� ; s z"SubclassedCategorical._constructorN)r� r� r� r� r� r� r� r� r� r� : s r� c s$ �r 176 �fdd�}|S � fdd�}|S )a` 177 Create a function for calling on an array. 178 179 Parameters 180 ---------- 181 alternative : function 182 The function to be called on the array with no NaNs. 183 Only used when 'skipna_alternative' is None. 184 skipna_alternative : function 185 The function to be called on the original array 186 187 Returns 188 ------- 189 function 190 c s 191 � | j �S r� )r� �r� )�skipna_alternativer� r� �skipna_wrapperR r+ z,_make_skipna_wrapper.<locals>.skipna_wrapperc s" | � � }t|�dkr tjS � |�S )Nr )�dropnar r� r� )r� Znona)� alternativer� r� r� W s r� )r� r� r� r� )r� r� r� �_make_skipna_wrapper@ s 192 �r� � rows_list� list[str]c C s t j}|�| �| S )aW 193 Convert list of CSV rows to single CSV-formatted string for current OS. 194 195 This method is used for creating expected value of to_csv() method. 196 197 Parameters 198 ---------- 199 rows_list : List[str] 200 Each element represents the row of csv. 201 202 Returns 203 ------- 204 str 205 Expected output of to_csv() in current OS. 206 )r� �linesep�join)r� �sepr� r� r� � convert_rows_list_to_csv_str` s r� �expected_exception�type[Exception]r c C s ddl }|j| dd�S )a$ 207 Helper function to mark pytest.raises that have an external error message. 208 209 Parameters 210 ---------- 211 expected_exception : Exception 212 Expected error to raise. 213 214 Returns 215 ------- 216 Callable 217 Regular `pytest.raises` function with `match` equal to `None`. 218 r N)�match)�pytest�raises)r� r� r� r� r� �external_error_raisedt s r� c s@ g }|D ]\�� |� ��� f� |� ��fdd�tD �7 }q|S )a� 219 Combine frame, functions from com._cython_table 220 keys and expected result. 221 222 Parameters 223 ---------- 224 ndframe : DataFrame or Series 225 func_names_and_expected : Sequence of two items 226 The first item is a name of a NDFrame method ('sum', 'prod') etc. 227 The second item is the expected return value. 228 229 Returns 230 ------- 231 list 232 List of three items (DataFrame, function, expected result) 233 c s" g | ] \}}|�kr�|� f�qS r� r� )r� �funcr� �r� � func_name�ndframer� r� r� � s 234 �z+get_cython_table_params.<locals>.<listcomp>)rk � cython_table)r� Zfunc_names_and_expected�resultsr� r� r� �get_cython_table_params� s 235 �r� �op_namer c sN | � d�}ztt|�}W |S ty& tt|dd� �� � fdd�}Y |S w )z� 236 The operator function for a given op name. 237 238 Parameters 239 ---------- 240 op_name : str 241 The op name, in form of "add" or "__add__". 242 243 Returns 244 ------- 245 function 246 A function performing the operation. 247 r\ r� Nc s 248 � || �S r� r� )r� �y��ropr� r� r{ � s 249 z"get_op_from_name.<locals>.<lambda>)�stripr� �operator�AttributeError)r� Z short_opname�opr� r� r� �get_op_from_name� s 250 ��r� c C � | S r� r� r� r� r� r� �getitem� � r� c C r� r� r� r� r� r� r� �setitem� r� r� c C � | j S r� )�locr� r� r� r� r� � � r� c C r� r� )�ilocr� r� r� r� r� � r� r� c C r� r� )�atr� r� r� r� r� � r� r� c C r� r� )�iatr� r� r� r� r� � r� r� c C s� t | tj�rt |tj�rt�| |�S t | tj�r t|| �S t | t�r$dS t | t�r/t| j|�S t | ttf�r<t| j |�S t | t 251 �rGt| j |�S t | t j jj�rUt| j|�S t | t j jj�rit| j|�pht| j|�S t | t�r�| jdkr�t |t�r�|jdkr�| j}|j}|�d��� d }|�d��� d }||kS t | t�r�t |t�r�t�| j|j�p�t�| j|j�S t | t�r�t | j j�dkr�| j jd }t||�S t t | �t |���)z- 252 Pandas-compat for np.shares_memory. 253 Fzstring[pyarrow]r r� ) r� r� r� � shares_memoryr r �_codesr r r� rR �_ndarrayr� �core�arrays� SparseArray� sp_values� IntervalArray�_left�_rightrP r� �_data�chunk�buffersrO �_maskr r �_mgrr� �type)�left�rightZ left_pa_dataZ right_pa_dataZ left_buf1Z 254 right_buf1�arrr� r� r� r� � s> 255 256 257 258 � 259 r� )y�ALL_INT_EA_DTYPES�ALL_INT_NUMPY_DTYPES�ALL_NUMPY_DTYPES�ALL_REAL_NUMPY_DTYPESr r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r( r= r> r? r� r{ r� r} r r rt r� rx rD �EMPTY_STRING_PATTERN�ENDIANrE rF rG r� r� rq rn r� r� rA r� r rK rB r� r= r4 r; r� r� r r� r� r� r� rx r? r r r* r� r� r� rL r r� r- rM r r7 r& r� r� r, r> r r6 r� r) �NARROW_NP_DTYPESr �NP_NAT_OBJECTS� NULL_OBJECTSr r@ r% r& r� r� rC r! r" r# r� r r� rH r� rk rf rw r� r� r� ry r� ra r\ rI rJ r$ )r� r� )r� rz )T)r� ru )r� N)r� r )r� r� N)r� r )r� r )r� rM )r� rL )r� rN )r� r )r� rK )r� r N)r� r r� r )r� r� N)r� r r� rU )r� r r� rT )r� )r� r r� r )r� r r� )r� r. )Nr N)NN)Nr )r� r )rN FNN)r rO r� r ) 260 TTr� r� NNNNNN)r� N)r� r� r� ru )r� r� r� r )r� ru r� r )�� 261 __future__r rh r �decimalr r� r� r] r� �sysr �typingr r r r r 262 r� �numpyr� �pandas._config.localizationr r r �pandas._typingr � pandas.compatr �pandas.core.dtypes.commonr r r r r �pandasr� r r r r r r r r r r �pandas._testing._ior r r! r" r# r$ �pandas._testing._randomr% r& r' �pandas._testing._warningsr( r) �pandas._testing.assertersr* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r@ �pandas._testing.compatrA rB �pandas._testing.contextsrC rD rE rF rG rH rI rJ �pandas.core.apirK rL rM rN �pandas.core.arraysrO rP rQ �pandas.core.arrays._mixinsrR �pandas.core.constructionrS rT rU r! r3 r\ �__annotations__ra r rf rk r� r� �floatrn rq �complexrt ru rw rx ry rz r{ r| r} r~ r r� r� �float16rl rb rc rd rX rY rZ r� r� r� r� �NAr� � 263 datetime64� timedelta64r� �pyarrowr� r[ �UNSIGNED_INT_PYARROW_DTYPESre �SIGNED_INT_PYARROW_DTYPES�ALL_INT_PYARROW_DTYPESrm �FLOAT_PYARROW_DTYPES�utf8�STRING_PYARROW_DTYPESZtime32Ztime64�TIME_PYARROW_DTYPESZdate32Zdate64�DATE_PYARROW_DTYPES�DATETIME_PYARROW_DTYPES�TIMEDELTA_PYARROW_DTYPES�bool_�BOOL_PYARROW_DTYPES�ALL_PYARROW_DTYPES�compiler� �DeprecationWarning�ResourceWarningr� 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- r4 r6 r7 r; r= r> r? rK rL rM rx r� r� r� r� r� r� r� r� r� r� �common� _cython_table�itemsr� r� r� r� r� r� r� r� r� r� �__all__r� r� r� r� �<module> sp 0 d( 264 �������� � 265 � �� 266 ������� 267 268 269 270 271 272 ) 273 274 275 276 277 �u � 278 k 279 280 1