
    .i                     \    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      Zy)    )methodcallerN)
MultiIndexSeries
date_rangec                   p   e Zd Zej                  j                  dddg      d        Zd Zd Zej                  j                  de	j                  ej                  dd	g      d
        Zej                  j                  dddg      d        Zej                  j                  dg d      d        Zd Zd Zy)
TestSeriesfuncrename_axis_set_axis_namec           
      z   t        g dt        j                  dD cg c]  }d|f c}ddg            } t        |dd	g      |      }|j                  j
                  J |j                  j                  ddgk(  sJ |j                  j
                  J |j                  j                  s	J dd	g       y c c}w )
N)         )aBcAl1l2)names)indexL1L2)r   r   from_tuplesr   r   namer   )selfr	   xserresults        ^/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/tests/generic/test_series.pytest_set_axis_name_miz TestSeries.test_set_axis_name_mi   s    ((#23a#q3D$<
 2dT4L1#6yy~~%%%yy4,...||  (((||!!/D$</! 4s   B8c                     t        dg      }d}t        j                  t        |      5  |j	                  dd       d d d        y # 1 sw Y   y xY w)N   z&No axis named 1 for object type Seriesmatchr   )r   axis)r   pytestraises
ValueErrorr   )r   r   msgs      r    test_set_axis_name_raisesz$TestSeries.test_set_axis_name_raises   sH    aSk6]]:S1 	1Ca0	1 	1 	1s   AAc                 j    t        g d      }|j                         }t        j                  ||       y )N)TFT)r   _get_bool_datatmassert_series_equal)r   r   r   s      r    !test_get_bool_data_preserve_dtypez,TestSeries.test_get_bool_data_preserve_dtype%   s+    ()##%
vs+    dataTFc                     t        |g      }d}t        j                  t        |      5  t	        |       d d d        y # 1 sw Y   y xY wNz(The truth value of a Series is ambiguousr$   r   r'   r(   r)   bool)r   r2   seriesr*   s       r    #test_nonzero_single_element_raise_1z.TestSeries.test_nonzero_single_element_raise_1*   s>     8]]:S1 	L	 	 		   ?A)TT)FFc                     d}t        |g      }t        j                  t        |      5  t	        |       d d d        y # 1 sw Y   y xY wr4   r5   )r   r2   msg_errr7   s       r    #test_nonzero_multiple_element_raisez.TestSeries.test_nonzero_multiple_element_raise3   s>     =]]:W5 	L	 	 	r9   )r#   r   r   g        c                     d}t        |g      }t        j                  t        |      5  t	        |       d d d        y # 1 sw Y   y xY wr4   r5   )r   r2   msg_err1r7   s       r    !test_nonbool_single_element_raisez,TestSeries.test_nonbool_single_element_raise;   s>     >]]:X6 	L	 	 	r9   c                    t        t        j                  j                  d      j                  d      t	        ddd      d      }|j                  d      j                         }t        j                  ||       |j                  d      j                         }t        j                  ||       |j                  d      j                  d	       }t        j                  ||       y )
N   i  20130101s)periodsfreqfoo)r   r   1minc                 "    | j                         S N)sum)r   s    r    <lambda>zETestSeries.test_metadata_propagation_indiv_resample.<locals>.<lambda>P   s    QUUW r1   )r   nprandomdefault_rngr   resamplemeanr.   assert_metadata_equivalentminapply)r   tsr   s      r    (test_metadata_propagation_indiv_resamplez3TestSeries.test_metadata_propagation_indiv_resampleC   s    II!!!$++D1ZC@

 V$))+
%%b&1V$((*
%%b&1V$**+<=
%%b&1r1   c                    t        t        d      t        d            }d|_        t        t        d      t        d            }d|_        |j                  }t	        j
                  ||       d
d}|j                         5 }|j                  t         dddg       |j                  t         d|       d|_        d|_        t        j                  ||g      }|j                  d	k(  sJ |j                  J 	 d d d        y # 1 sw Y   y xY w)N   rF   barc                 :   | j                   D ]  }|dk(  r]|dk(  rXdj                  |j                  D cg c]  }t        ||d       rt        ||       c}      }t        j                  | ||       et        j                  | |t        ||d               | S c c}w )Nconcatfilename+)	_metadatajoin
input_objsgetattrobject__setattr__)r   othermethodkwargsr   objvalues          r    finalizez<TestSeries.test_metadata_propagation_indiv.<locals>.finalize^   s     OX%$**<HH (-'7'7 #&sD$7 $C.E &&tT59&&tT75$3MNO Ks    B
r]   r   r[   __finalize__zfoo+barrI   )r   ranger   Tr.   rQ   contextsetattrr[   pdrZ   )r   monkeypatchr   ser2r   rh   ms          r    test_metadata_propagation_indivz*TestSeries.test_metadata_propagation_indivS   s     U1XuQx(eAha)	
%%c62	    " 		'aIIfkFJ+?@IIfnh7 CL!DMYYT{+F??i///;;&&&		' 		' 		's   A5D  D	N)__name__
__module____qualname__r'   markparametrizer!   r+   r0   rL   nanrn   NaTr8   r<   r?   rU   rr    r1   r    r   r      s    [[Vm5E%FG0 H01,
 [[VbffbffdE%BC D [[VlN%CD E [[V%56 72 $'r1   r   )operatorr   numpyrL   r'   pandasrn   r   r   r   pandas._testing_testingr.   r   rz   r1   r    <module>r      s,    !    
 h' h'r1   