
    g
                     @    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       T/var/www/openai/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	        |      }d}t        j                  t        |d      5  t        j                  |      }d d d        t        j                  t        |      5  |j                  d       }d d d        t        j                         y # 1 sw Y   VxY w# 1 sw Y   ,xY w)Nr	   )      z:The behavior of DataFrame.sum with axis=None is deprecatedF)matchcheck_stacklevel)r+   )axis)
r   randomdefault_rngstandard_normalr   r   assert_produces_warningFutureWarningsumassert_series_equal)r   arrr   msgresr   s         r   !test_sum_deprecated_axis_behaviorz-TestAsArray.test_sum_deprecated_axis_behavior   s     ii##A&66v>s^J''u
 &&*C

 ''SAvv4v(H B
sH-
 

 BAs   CCCC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*   )r:   r;   r<   r=   r>   r?   r   )x1x2x3)r#   )r   r   ravelr   reshaper   r   )r   r5   batchr   r   s        r   test_np_ravelzTestAsArray.test_np_ravel/   s$   hh56
 sKse9U]]1a%89sKL88	
 	##FH5)CFNN1a$8BTUV88AB
##FH5 ! E %--1-7IJ 
 88	
 	##FH5C L$s   "E:&EN)__name__
__module____qualname__r   r'   r8   rF    r   r   r   r      s    6B."*6r   r   )
__doc__numpyr   pandasr   r   pandas._testing_testingr   r   rJ   r   r   <module>rP      s%     L6 L6r   