
    .i                     <    d dl Zd dlZd dlZd dlmZ  G d d      Zy)    Nc            
           e Zd Zej                  j                  d ej                  dd       ej                  ddd       ej                  dd       ej                  dd      g      d        Z
d	 Zd
 Zy)TestCombinedata2000   )periodsz
US/Central)r   tzr   c                     t        j                  d|i      }|j                         }d |j                  d<   d }|j	                  ||      }t        j                  ||       y )NA)   r   c                     |S N abs     e/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/tests/frame/methods/test_combine.pycombinerz8TestCombine.test_combine_datetlike_udf.<locals>.combiner       H    )pd	DataFramecopyiloccombinetmassert_frame_equal)selfr   dfotherr   results         r   test_combine_datetlike_udfz&TestCombine.test_combine_datetlike_udf	   sU     \\3+&		 E8,
fe,r   c                    |}|j                   |j                  d d g df   }|j                  |t        j                        }|j                  |t        j                        }|d   j                         j                         sJ |d   j                         j                         sJ |j                   |j                  d d g df   }|j                   |j                  d d g df   }|j                   |j                  d d g df   j                  |      dz  }t        j                  ||       t        j                  ||       y )N)r   BCD   )
locindexr   npaddisnaallreindex_liker   r   )	r   float_framedf1df2combined	combined2chunkchunk2exps	            r   test_combine_genericz TestCombine.test_combine_generic   s$   ook//4oEF;;sBFF+KKRVV,	}!!#'')))~""$((***X^^CR0/ABys3_DE OOK--cr2OCDQQRWX 	 	eS)
fc*r   c                 (   t        j                  t        d      t        d      d      }ddg|_        |j	                         }d |j
                  dd d f<   d }|j                  ||      }|j                  d      }t        j                  ||       y )N   )r   r%   r   r   c                     |S r   r   r   s     r   r   z<TestCombine.test_combine_nonunique_columns.<locals>.combiner:   r   r   float64)
r   r   rangecolumnsr   r   r   astyper   r   )r   r   r    r   r!   expecteds         r   test_combine_nonunique_columnsz*TestCombine.test_combine_nonunique_columns1   s{     \\auQx893Z
	1	 E8,<<	*
fh/r   N)__name__
__module____qualname__pytestmarkparametrizer   
date_rangeperiod_rangetimedelta_ranger"   r8   rA   r   r   r   r   r      sy    [[BMM&!,BMM&!=BOOFA.Bq!,		

-
-+&0r   r   )	numpyr+   rE   pandasr   pandas._testing_testingr   r   r   r   r   <module>rO      s       70 70r   