o
    ZZŽhK  ã                   @   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   ÚpytestZraisesÚ
ValueErrorÚmT)ÚmsgÚma_arr© r   úN/var/www/auris/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Ú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    