
    Pi	                     H    d Z ddlZddlmZmZ ddlmZ  G d d          Z	dS )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dS )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        ||           d S )N      )ABobject)dtype)r   r   npasarrayarraytmassert_numpy_array_equal)selfdfresultexpecteds       s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/frame/test_npfuncs.pytest_asarray_homogeneousz$TestAsArray.test_asarray_homogeneous   sv    [!Q00{Aq67J7JKKLLB8aVaV,H===
#FH55555    c                    t          j        d          5  t          j        |          }d d d            n# 1 swxY w Y   t          |t	          |                    sJ |j                            |j                  sJ |j                            |j                  sJ t          j	        ||
                    t           j                             d S )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&&& 	* 	*W[))F	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	*&${"3"344444| 122222~!!+"566666
fk&7&7&@&@AAAAAs   7;;c                    t           j                            d                              d          }t	          |          }t          j        |          }|                                                    d           }||k    sJ d S )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   sn     i##A&&66v>>s^^fRjj;;==$$$$//hr   c                 2   t          j        g dg dg          }t          j        d |D                       }t          j        g d          }t          j        ||           t          j        t          |d                             dd          g d	                    }t          j        g d          }t          j        ||           t          j        d
 |D                       }t          j        g d          }t          j        ||           d S )N)=w?r2d?B^)Ҫ?T2o灨$пc                 T    g | ]%}t          |                    d d                    &S )r   r*   r   reshape.0batchs     r   
<listcomp>z-TestAsArray.test_np_ravel.<locals>.<listcomp>3   s.    KKKe9U]]1a%8%899KKKr   )r5   r6   r7   r8   r9   r:   r   r   r*   x1x2x3r#   c                 \    g | ])}t          |                    d d          g d          *S )r   r*   rB   rF   r<   r>   s     r   rA   z-TestAsArray.test_np_ravel.<locals>.<listcomp>E   sJ        %--1--7I7I7IJJJ  r   )r   r   ravelr   r   r   r=   )r   r1   r   r   s       r   test_np_ravelzTestAsArray.test_np_ravel*   sG   h555666
 
 KKsKKKLL8  	
 	
 	#FH555)CFNN1a$8$8BTBTBTUUUVV8AAABB
#FH555    
 
 8  	
 	
 	#FH55555r   N)__name__
__module____qualname__r   r'   r3   rI    r   r   r   r      sS        6 6 6B B B	 	 	*6 *6 *6 *6 *6r   r   )
__doc__numpyr   pandasr   r   pandas._testing_testingr   r   rM   r   r   <module>rS      s                     F6 F6 F6 F6 F6 F6 F6 F6 F6 F6r   