a
    Pf                     @   s   d dl Zd dlZd dlmZmZmZ d dlmZ	 d dl
mZ ejjddd dd gdd	gd
dd Zejjddd dd gdd	gd
dd Zdd Zejdddgdd Zdd Zdd Zdd Zdd Zd d! Zd"d# ZdS )$    N)	DataFrameSeries
date_range)	get_arraymethodc                 C   s   | j S Nvaluesser r   Z/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/copy_view/test_array.py<lambda>       r   c                 C   s
   t | S r   npasarrayr
   r   r   r   r      r   r	   r   )Zidsc                 C   s   t g ddd}| }||}| rt|t|ds:J |jjdu sJJ tjt	dd d|d< W d    n1 sv0    Y  t
|| d|jd< |jd dksJ n*|jjdu sJ d|d< |jd dksJ d S )	N         namer   F	read-onlymatchr   T)r   copyr   shares_memoryr   flags	writeablepytestraises
ValueErrortmassert_series_equalilocr	   )using_copy_on_writer   r   ser_origarrr   r   r   test_series_values   s    &
r)   c                 C   s   | j S r   r   dfr   r   r   r   0   r   c                 C   s
   t | S r   r   r*   r   r   r   r   0   r   c                 C   s   t g dg dd}| }||}| rt|t|ds@J |jjdu sPJ tjt	dd d|d	< W d    n1 s|0    Y  t
|| d|jd	< |jd	 dksJ n<|jjd
u sJ d|d	< |s|jd	 dksJ nt
|| d S )Nr   )         abr0   Fr   r   r   )r   r   T)r   r   r   r   r   r   r   r    r!   r"   r#   Zassert_frame_equalr%   r	   )r&   using_array_managerr   r+   Zdf_origr(   r   r   r   test_dataframe_values.   s     &
r3   c                 C   sd  t g ddd}| }| }| rt|t|ds:J |jjdu sJJ tj	t
dd d|d< W d    n1 sv0    Y  t|| d|jd< |jd dksJ n*|jjdu sJ d|d< |jd dksJ t g ddd}|jdd	}t|t|drJ |jjdu sJ t g ddd}|jd
d}t|t|drNJ |jjdu s`J d S )Nr   r   r   Fr   r   r   T)r   Zfloat64Zdtype)r   r   Zto_numpyr   r   r   r   r   r    r!   r"   r#   r$   r%   r	   )r&   r   r'   r(   r   r   r   test_series_to_numpyO   s,    &
r5   orderFCc                 C   sr   t g d}tjtdd |j|d}W d    n1 s<0    Y  | rZ|jjdu sZJ tt	||snJ d S )Nr   zis deprecatedr   )r6   F)
r   r#   Zassert_produces_warningFutureWarningZravelr   r   r   r   r   )r&   r6   r   r(   r   r   r   test_ravel_read_onlyt   s    *r:   c                 C   s   t g ddd}tj|dd}t|t|s2J | rH|jjdu sXJ n|jjdu sXJ t|}t|t|svJ | r|jjdu sJ n|jjdu sJ d S )Nr   Int64r4   int64FT)r   r   r   r   r   r   r   )r&   r   r(   r   r   r   test_series_array_ea_dtypes~   s    
r=   c                 C   s   t dg didd}tj|dd}t|t|ds8J | rN|jjdu s^J n|jjdu s^J t|}t|t|ds~J | r|jjdu sJ n|jjdu sJ d S )Nr0   r   r;   r4   r<   FTr   r   r   r   r   r   r   r&   r+   r(   r   r   r   test_dataframe_array_ea_dtypes   s    
r@   c                 C   sb   t dddgidd}t|}|s8t|t|ds8J | rN|jjdu s^J n|jjdu s^J d S )Nr0   r1   stringr4   FTr>   )r&   r2   r+   r(   r   r   r   !test_dataframe_array_string_dtype   s    
rB   c                  C   sF   t g ddd} t| }t|t| dr2J |jjdu sBJ d S )Nr   g      ?r/   r0   Tr>   r+   r(   r   r   r   $test_dataframe_multiple_numpy_dtypes   s    
rD   c                 C   sH   t dtdddi}t|}| r4|jjdu sDJ n|jjdu sDJ d S )Nr0   z
2012-01-01r   )ZperiodsFT)r   r   r   r   r   r   r?   r   r   r   test_values_is_ea   s
    
rE   c                  C   s$   t  } t| }|jjdu s J d S )NT)r   r   r   r   r   rC   r   r   r   test_empty_dataframe   s    
rF   )numpyr   r    Zpandasr   r   r   Zpandas._testingZ_testingr#   Zpandas.tests.copy_view.utilr   markZparametrizer)   r3   r5   r:   r=   r@   rB   rD   rE   rF   r   r   r   r   <module>   s2   

%
		