REDROOM

Edit File Size: 2.37 KB

/opt/alt/python37/lib64/python3.7/site-packages/numpy/ma/tests/test_deprecations.py

Text Base64

"""Test deprecation and future warnings.

"""
from __future__ import division, absolute_import, print_function

import numpy as np
from numpy.testing import TestCase, run_module_suite, assert_warns
from numpy.ma.testutils import assert_equal
from numpy.ma.core import MaskedArrayFutureWarning

class TestArgsort(TestCase):
    """ gh-8701 """
    def _test_base(self, argsort, cls):
        arr_0d = np.array(1).view(cls)
        argsort(arr_0d)

arr_1d = np.array([1, 2, 3]).view(cls)
        argsort(arr_1d)

# argsort has a bad default for >1d arrays
        arr_2d = np.array([[1, 2], [3, 4]]).view(cls)
        result = assert_warns(
            np.ma.core.MaskedArrayFutureWarning, argsort, arr_2d)
        assert_equal(result, argsort(arr_2d, axis=None))

# should be no warnings for explicitly specifying it
        argsort(arr_2d, axis=None)
        argsort(arr_2d, axis=-1)

def test_function_ndarray(self):
        return self._test_base(np.ma.argsort, np.ndarray)

def test_function_maskedarray(self):
        return self._test_base(np.ma.argsort, np.ma.MaskedArray)

def test_method(self):
        return self._test_base(np.ma.MaskedArray.argsort, np.ma.MaskedArray)

class TestMinimumMaximum(TestCase):
    def test_minimum(self):
        assert_warns(DeprecationWarning, np.ma.minimum, np.ma.array([1, 2]))

def test_maximum(self):
        assert_warns(DeprecationWarning, np.ma.maximum, np.ma.array([1, 2]))

def test_axis_default(self):
        # NumPy 1.13, 2017-05-06

data1d = np.ma.arange(6)
        data2d = data1d.reshape(2, 3)

ma_min = np.ma.minimum.reduce
        ma_max = np.ma.maximum.reduce

# check that the default axis is still None, but warns on 2d arrays
        result = assert_warns(MaskedArrayFutureWarning, ma_max, data2d)
        assert_equal(result, ma_max(data2d, axis=None))

result = assert_warns(MaskedArrayFutureWarning, ma_min, data2d)
        assert_equal(result, ma_min(data2d, axis=None))

# no warnings on 1d, as both new and old defaults are equivalent
        result = ma_min(data1d)
        assert_equal(result, ma_min(data1d, axis=None))
        assert_equal(result, ma_min(data1d, axis=0))

result = ma_max(data1d)
        assert_equal(result, ma_max(data1d, axis=None))
        assert_equal(result, ma_max(data1d, axis=0))

if __name__ == "__main__":
    run_module_suite()

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2023-05-13 02:18:14	Edit Download recursive recursive
test_core.py	177.42 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_deprecations.py	2.37 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_extras.py	62.43 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_mrecords.py	20.05 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_old_ma.py	32.56 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_regression.py	2.51 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive
test_subclassing.py	13.63 KB	lrw-r--r--	2023-04-24 19:30:09	Edit Download recursive recursive

Directory Contents