File: //lib/python3/dist-packages/numpy/ma/tests/__pycache__/test_extras.cpython-310.pyc
o
6��a� � @ s� d Z ddlZddlZddlZddlZddlmZmZ ddl m
Z
mZmZm
Z
ddlmZmZmZmZmZmZmZmZmZmZmZ ddlmZmZmZmZmZm 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; G dd� d�Z<G d d
� d
�Z=G dd� d�Z>G d
d� d�Z?G dd� 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 d!d"� d"�ZIdS )#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)�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�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� ��maskr8 �a�b�f��names�formats�r r �r9 r9 �r6 r6 �r<