
    bMh
                     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       _/var/www/html/test/jupyter/venv/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	          |          }d}t          j        t          |d          5  t          j        |          }d d d            n# 1 swxY w Y   t          j        t          |          5  |                    d           }d d d            n# 1 swxY w Y   t          j	        ||           d S )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   sS    i##A&&66v>>s^^J'u
 
 
 	 	 &**C	 	 	 	 	 	 	 	 	 	 	 	 	 	 	
 'SAAA 	) 	)vv4v((H	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	)
sH-----s$    BBB'C

CC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>8   s.    KKKe9U]]1a%8%899KKKr   )r:   r;   r<   r=   r>   r?   r   r   r*   x1x2x3r#   c                 \    g | ])}t          |                    d d          g d          *S )r   r*   rG   rK   rA   rC   s     r   rF   z-TestAsArray.test_np_ravel.<locals>.<listcomp>J   sJ        %--1--7I7I7IJJJ  r   )r   r   ravelr   r   r   rB   )r   r5   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'   r8   rN    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   rR   r   r   <module>rX      s                     L6 L6 L6 L6 L6 L6 L6 L6 L6 L6r   