a
    d0                     @   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   j/var/www/html/stable-diffusion-webui/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   )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br1   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     &
r4   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   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,    &
r7   orderFCc                 C   sD   t g d}|j|d}| r,|jjdu s,J tt||s@J d S )Nr   )r8   F)r   Zravelr   r    r   r   r   )r'   r8   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|rvJ |jjdu sJ d S )Nr   Int64r6   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r8J |jjdu sHJ t|}| rh|jjdu sxJ n|jjdu sxJ d S )Nr1   r   r<   r6   r=   TFr   r   r   r   r   r   r    r'   r,   r)   r   r   r   test_dataframe_array_ea_dtypes   s    
rA   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 )Nr1   r2   stringr6   FTr?   )r'   r3   r,   r)   r   r   r   !test_dataframe_array_string_dtype   s    
rC   c                  C   sF   t g ddd} t| }t|t| dr2J |jjdu sBJ d S )Nr   g      ?r0   r1   Tr?   r,   r)   r   r   r   $test_dataframe_multiple_numpy_dtypes   s    
rE   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 )Nr1   z
2012-01-01r   )ZperiodsFT)r   r   r   r   r   r    r@   r   r   r   test_values_is_ea   s
    
rF   c                  C   s$   t  } t| }|jjdu s J d S )NT)r   r   r   r   r    rD   r   r   r   test_empty_dataframe   s    
rG   )numpyr   r!   Zpandasr   r   r   Zpandas._testingZ_testingr$   Zpandas.tests.copy_view.utilr   markZparametrizer*   r4   r7   r;   r>   rA   rC   rE   rF   rG   r   r   r   r   <module>   s2   

%
	