| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465 |
- """Test deprecation and future warnings.
- """
- import pytest
- import numpy as np
- from numpy.ma.core import MaskedArrayFutureWarning
- from numpy.ma.testutils import assert_equal
- class TestArgsort:
- """ 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 = pytest.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:
- 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 = pytest.warns(MaskedArrayFutureWarning, ma_max, data2d)
- assert_equal(result, ma_max(data2d, axis=None))
- result = pytest.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))
|
