a :jgK@sVddlZddlZddlmZddlmZddZddZdZ ej d e d d Z dS) N) masked_array)assert_array_equalcCsPd}tgdgdd}tjt|d|jWdn1sB0YdS)Nz+matrix transpose with ndim < 2 is undefined))rrrrrrdatamask)match)rpytestZraises ValueErrormT)msgma_arrrI/usr/local/lib/python3.9/site-packages/numpy/ma/tests/test_arrayobject.py)test_matrix_transpose_raises_error_for_1ds rcCs6tgdgdggdgdgd}t|j|jdS)N)rrr)rrr )rrr)rrrr )rrTr)rrrr)test_matrix_transpose_equals_transpose_2dsr))rr)rrr)rrrrshapecCsrt|}t|d}t||}tjd}|jddg|d}t||d}t||d|d}t ||j dS)N*rr)sizer r) lennpZarangeZ broadcast_torandomZ default_rngchoicerZswapaxesrr)rZ num_of_axesZvecZarrrngr rZtgtrrr%test_matrix_transpose_equals_swapaxess   r!) rnumpyrZnumpy.marZ numpy.testingrrrZARRAY_SHAPES_TO_TESTmarkZ parametrizer!rrrrs