a
    Pf-                     @   sV   d dl mZ d dlZd dlZd dlZd dlmZmZm	Z	 d dl
mZ G dd dZdS )    )methodcallerN)
MultiIndexSeries
date_rangec                   @   s   e Zd Zejdddgdd Zdd Zdd	 Zd
d Z	ejde
jejddgdd Zejde
jejgdd Zejdddgdd Zejdg ddd Zdd Zdd ZdS )
TestSeriesfuncZrename_axis_set_axis_namec                 C   s   t g dtjdd dD ddgdd}t|d	d
g|}|jjd u sJJ |jjddgks^J |jjd u snJ |jjsJ d	d
gd S )N)         c                 S   s   g | ]}d |fqS )A ).0xr   r   Y/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/generic/test_series.py
<listcomp>       z4TestSeries.test_set_axis_name_mi.<locals>.<listcomp>)aBcl1l2)names)indexZL1ZL2)r   r   from_tuplesr   r   namer   )selfr   serresultr   r   r   test_set_axis_name_mi   s    z TestSeries.test_set_axis_name_mic                 C   sN   t dg}d}tjt|d |jddd W d    n1 s@0    Y  d S )N   z&No axis named 1 for object type Seriesmatchr   )r   Zaxis)r   pytestraises
ValueErrorr   )r   r   msgr   r   r   test_set_axis_name_raises   s    
z$TestSeries.test_set_axis_name_raisesc                 C   s$   t g d}| }t|| d S )N)TFT)r   Z_get_bool_datatmZassert_series_equal)r   r   r   r   r   r   !test_get_bool_data_preserve_dtype%   s    z,TestSeries.test_get_bool_data_preserve_dtypec                 C   s   d}t dg}t dg}tjt|d | s4J W d    n1 sH0    Y  tjt|d | rnJ W d    n1 s0    Y  d S )NMSeries.bool is now deprecated and will be removed in future version of pandasTFr!   )r   r(   assert_produces_warningFutureWarningbool)r   msg_warnr   Zser1r   r   r   test_nonzero_single_element*   s    

*z&TestSeries.test_nonzero_single_elementdataTFc                 C   sH   t |g}d}tjt|d t| W d    n1 s:0    Y  d S )N(The truth value of a Series is ambiguousr!   )r   r#   r$   r%   r-   )r   r0   seriesr&   r   r   r   #test_nonzero_single_element_raise_17   s    
z.TestSeries.test_nonzero_single_element_raise_1c              	   C   sz   d}d}t |g}tjt|dF tjt|d |  W d    n1 sN0    Y  W d    n1 sl0    Y  d S )Nr*   6bool cannot act on a non-boolean single element Seriesr!   )r   r(   r+   r,   r#   r$   r%   r-   r   r0   r.   Zmsg_errr2   r   r   r   #test_nonzero_single_element_raise_2@   s    
z.TestSeries.test_nonzero_single_element_raise_2)TT)FFc              	   C   s   d}d}t |g}tjt|d t| W d    n1 s>0    Y  tjt|dF tjt|d |  W d    n1 s0    Y  W d    n1 s0    Y  d S )Nr*   r1   r!   r   r#   r$   r%   r-   r(   r+   r,   r5   r   r   r   #test_nonzero_multiple_element_raiseL   s    
&z.TestSeries.test_nonzero_multiple_element_raise)r    r   r   g        c              	   C   s   d}d}d}t |g}tjt|d t| W d    n1 sB0    Y  tjt|dF tjt|d |  W d    n1 s0    Y  W d    n1 s0    Y  d S )Nr*   r1   r4   r!   r7   )r   r0   r.   Zmsg_err1Zmsg_err2r2   r   r   r   !test_nonbool_single_element_raise[   s    
&z,TestSeries.test_nonbool_single_element_raisec                 C   s~   t tjddtdddddd}|d }t|| |d	 }t|| |d
d	d
 }t|| d S )N   i  Z20130101s)Zperiodsfreqfoo)r   r   Z1minc                 S   s   |   S N)sum)r   r   r   r   <lambda>x   r   zETestSeries.test_metadata_propagation_indiv_resample.<locals>.<lambda>)r   nprandomZdefault_rngr   Zresamplemeanr(   assert_metadata_equivalentminapply)r   tsr   r   r   r   (test_metadata_propagation_indiv_resamplek   s    z3TestSeries.test_metadata_propagation_indiv_resamplec                 C   s   t tdtd}d|_t tdtd}d|_|j}t|| ddd}| f}|t dddg |t d	| d|_d|_t	
||g}|jd
ksJ |jd u sJ W d    n1 s0    Y  d S )N   r=   barc              	      s`   | j D ]T |dkrD dkrDd fdd|jD }t|  | qt|  t| d  q| S )Nconcatfilename+c                    s"   g | ]}t | d rt | qS r>   )getattr)r   objr   r   r   r      s   zPTestSeries.test_metadata_propagation_indiv.<locals>.finalize.<locals>.<listcomp>)	_metadatajoinZobjsobject__setattr__rN   )r   othermethodkwargsvaluer   rP   r   finalize   s    

z<TestSeries.test_metadata_propagation_indiv.<locals>.finalizerQ   r   rL   Z__finalize__zfoo+bar)N)r   ranger   Tr(   rD   contextsetattrrL   pdrK   )r   Zmonkeypatchr   Zser2r   rY   mr   r   r   test_metadata_propagation_indiv{   s    

z*TestSeries.test_metadata_propagation_indivN)__name__
__module____qualname__r#   markZparametrizer   r'   r)   r/   rA   nanr^   ZNaTr3   r6   r8   r9   rH   r`   r   r   r   r   r      s   




r   )operatorr   numpyrA   r#   Zpandasr^   r   r   r   Zpandas._testingZ_testingr(   r   r   r   r   r   <module>   s   