
    .i	                     @    d Z ddlZddlmZmZ ddlmZ  G d d      Z	y)z@
Tests for np.foo applied to DataFrame, not necessarily ufuncs.
    N)Categorical	DataFramec                   $    e Zd Zd Zd Zd Zd Zy)TestAsArrayc                     t        t        ddg      t        ddg      d      }t        j                  |      }t        j                  ddgddggd      }t        j                  ||       y )N      )ABobject)dtype)r   r   npasarrayarraytmassert_numpy_array_equal)selfdfresultexpecteds       ]/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/tests/frame/test_npfuncs.pytest_asarray_homogeneousz$TestAsArray.test_asarray_homogeneous   s]    [!Q0{Aq67JKLB88aVaV,H=
##FH5    c                    t        j                  d      5  t        j                  |      }d d d        t        t	        |            sJ |j
                  j                  |j
                        sJ |j                  j                  |j                        sJ t        j                  ||j                  t         j                               y # 1 sw Y   xY w)Nignore)all)r   errstatesqrt
isinstancetypeindexis_columnsr   assert_frame_equalapply)r   float_framer   s      r   test_np_sqrtzTestAsArray.test_np_sqrt   s    [[X& 	*WW[)F	*&${"3444|| 1 1222~~!!+"5"5666
fk&7&7&@A	* 	*s   CCc                     t         j                  j                  d      j                  d      }t	        |      }t        j
                  |      }|j                         j                  d       }||k(  sJ y )Nr	   )      )axis)r   randomdefault_rngstandard_normalr   sumto_numpy)r   arrr   resr   s        r   test_sum_axis_behaviorz"TestAsArray.test_sum_axis_behavior   s^     ii##A&66v>s^ffRj;;=$$$$/hr   c                    t        j                  g dg dg      }t        j                  |D cg c]  }t        |j	                  dd             c}      }t        j                  g d      }t        j                  ||       t        j                  t        |d   j	                  dd      g d            }t        j                  g d      }t        j                  ||       t        j                  |D cg c]!  }t        |j	                  dd      g d      # c}      }t        j                  g d      }t        j                  ||       y c c}w c c}w )	N)=w?r2d?B^)Ҫ?T2o灨$пr   r*   )r5   r6   r7   r8   r9   r:   r   )x1x2x3)r#   )r   r   ravelr   reshaper   r   )r   r1   batchr   r   s        r   test_np_ravelzTestAsArray.test_np_ravel*   s   hh56
 sKe9U]]1a%89KL88	
 	##FH5)CFNN1a$8BTUV88AB
##FH5 ! %--1-7IJ
 88	
 	##FH5C L$s   "E:&EN)__name__
__module____qualname__r   r'   r3   rA    r   r   r   r      s    6B	*6r   r   )
__doc__numpyr   pandasr   r   pandas._testing_testingr   r   rE   r   r   <module>rK      s%     F6 F6r   