a
    LfK                     @   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   sP   d} t g dg dd}tjt| d |j W d    n1 sB0    Y  d S )Nz+matrix transpose with ndim < 2 is undefined)                  )r   r   r   r   r   r   datamask)match)r   pytestZraises
ValueErrormT)msgma_arr r   X/var/www/ai-form-bot/venv/lib/python3.9/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npZarangeZbroadcast_torandomZdefault_rngchoicer   Zswapaxesr   r   )r   Znum_of_axesZvecZarrrngr   r   Ztgtr   r   r   %test_matrix_transpose_equals_swapaxes   s    r!   )r   numpyr   Znumpy.mar   Znumpy.testingr   r   r   ZARRAY_SHAPES_TO_TESTmarkZparametrizer!   r   r   r   r   <module>   s   