File: //usr/local/lib/python3.10/dist-packages/numpy/ma/tests/__pycache__/test_extras.cpython-310.pyc
o
���gj$ � @ s� d Z ddlZddlZddlZddlZddlmZ ddlm Z m
Z
ddlmZm
Z
mZmZ ddlmZmZmZmZmZmZmZmZmZmZmZ ddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z> G dd � d �Z?G d
d� d�Z@G dd
� d
�ZAG dd� d�ZBG dd� d�ZCG dd� d�ZDG dd� d�ZEG dd� d�ZFG dd� d�ZGG dd� d�ZHG dd� d�ZIG dd� d�ZJG d d!� d!�ZKG d"d#� d#�ZLG d$d%� d%�ZMdS )&z�Tests suite for MaskedArray.
Adapted from the original test_ma by Pierre Gerard-Marchant
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: test_extras.py 3473 2007-10-29 15:18:13Z jarrod.millman $
� N)�normalize_axis_tuple)�assert_warns�suppress_warnings)�assert_�assert_array_equal�assert_equal�assert_almost_equal)�array�arange�masked�MaskedArray�masked_array�getmaskarray�shape�nomask�ones�zeros�count)"�
atleast_1d�
atleast_2d�
atleast_3d�mr_�dot�polyfit�cov�corrcoef�median�average�unique�setxor1d� setdiff1d�union1d�intersect1d�in1d�ediff1d�apply_over_axes�apply_along_axis�compress_nd�compress_rowcols�mask_rowcols�clump_masked�clump_unmasked�flatnotmasked_contiguous�notmasked_contiguous�notmasked_edges�
masked_all�masked_all_like�isin�diagflat�ndenumerate�stack�vstackc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zdd� Zd
d� Z dd� Z
dS )�TestGenericc C sJ t dtd�}tddgddgtd�}t||� t�ddgddgd��}t d|d�}td d gd
d
g|d�}t||� t d|d�}td d gd d ggd
d
gd
d
gg|d�}t||� t�ddd
dgfg�}t d|d�}tddgddg|d�}t||� t d|d�}tddgddg|d�}t||� t d
|d�}tdggdgg|d�}t||� d S )N�� ��dtype� ��maskr: �a�b�f��names�formats�r r �r; r; �r8 r8 �r>