o
    gK                     @   sV   d dl Z d dlZd dlmZ d dlmZ dd Zdd ZdZ	e j
d	e	d
d ZdS )    N)masked_array)assert_array_equalc                  C   sR   d} t g dg dd}tjt| d |j W d    d S 1 s"w   Y  d S )Nz+matrix transpose with ndim < 2 is undefined)                  )r   r   r   r   r   r   datamask)match)r   pytestraises
ValueErrormT)msgma_arr r   Y/var/www/html/api-tag/env/lib/python3.10/site-packages/numpy/ma/tests/test_arrayobject.py)test_matrix_transpose_raises_error_for_1d   s   "r   c                  C   s6   t g dg dgg dg dgd} t| j| j d S )N)r   r   r   )r   r   r	   )r   r   r   )r   r   r   r
   )r   r   Tr   )r   r   r   r   )test_matrix_transpose_equals_transpose_2d   s   r   ))r   r   )r   r   r   )r   r   r   r   shapec                 C   sr   t | }t| d }t|| }tjd}|jddg| d}t||d}t||d |d }t	||j
 d S )N*   r   r   )sizer
   r   )lennparangebroadcast_torandomdefault_rngchoicer   swapaxesr   r   )r   num_of_axesvecarrrngr   r   tgtr   r   r   %test_matrix_transpose_equals_swapaxes   s   r*   )r   numpyr   numpy.mar   numpy.testingr   r   r   ARRAY_SHAPES_TO_TESTmarkparametrizer*   r   r   r   r   <module>   s    