o
    ðT·jK  ã                   @   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   úb/home/dinkstrade/pdmp-scanner/venv/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    