| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 |
- import numpy as np
- from numpy.testing import assert_, assert_array_equal
- class TestRegression:
- def test_masked_array_create(self):
- # Ticket #17
- x = np.ma.masked_array([0, 1, 2, 3, 0, 4, 5, 6],
- mask=[0, 0, 0, 1, 1, 1, 0, 0])
- assert_array_equal(np.ma.nonzero(x), [[1, 2, 6, 7]])
- def test_masked_array(self):
- # Ticket #61
- np.ma.array(1, mask=[1])
- def test_mem_masked_where(self):
- # Ticket #62
- from numpy.ma import MaskType, masked_where
- a = np.zeros((1, 1))
- b = np.zeros(a.shape, MaskType)
- c = masked_where(b, a)
- a - c
- def test_masked_array_multiply(self):
- # Ticket #254
- a = np.ma.zeros((4, 1))
- a[2, 0] = np.ma.masked
- b = np.zeros((4, 2))
- a * b
- b * a
- def test_masked_array_repeat(self):
- # Ticket #271
- np.ma.array([1], mask=False).repeat(10)
- def test_masked_array_repr_unicode(self):
- # Ticket #1256
- repr(np.ma.array("Unicode"))
- def test_atleast_2d(self):
- # Ticket #1559
- a = np.ma.masked_array([0.0, 1.2, 3.5], mask=[False, True, False])
- b = np.atleast_2d(a)
- assert_(a.mask.ndim == 1)
- assert_(b.mask.ndim == 2)
- def test_set_fill_value_unicode_py3(self):
- # Ticket #2733
- a = np.ma.masked_array(['a', 'b', 'c'], mask=[1, 0, 0])
- a.fill_value = 'X'
- assert_(a.fill_value == 'X')
- def test_var_sets_maskedarray_scalar(self):
- # Issue gh-2757
- a = np.ma.array(np.arange(5), mask=True)
- mout = np.ma.array(-1, dtype=float)
- a.var(out=mout)
- assert_(mout._data == 0)
- def test_mask_not_backmangled(self):
- # See gh-10314. Test case taken from gh-3140.
- a = np.ma.MaskedArray([1., 2.], mask=[False, False])
- assert_(a.mask.shape == (2,))
- b = np.tile(a, (2, 1))
- # Check that the above no longer changes a.shape to (1, 2)
- assert_(a.mask.shape == (2,))
- assert_(b.shape == (2, 2))
- assert_(b.mask.shape == (2, 2))
- def test_empty_list_on_structured(self):
- # See gh-12464. Indexing with empty list should give empty result.
- ma = np.ma.MaskedArray([(1, 1.), (2, 2.), (3, 3.)], dtype='i4,f4')
- assert_array_equal(ma[[]], ma[:0])
- def test_masked_array_tobytes_fortran(self):
- ma = np.ma.arange(4).reshape((2, 2))
- assert_array_equal(ma.tobytes(order='F'), ma.T.tobytes())
- def test_structured_array(self):
- # see gh-22041
- np.ma.array((1, (b"", b"")),
- dtype=[("x", np.int_),
- ("y", [("i", np.void), ("j", np.void)])])
|
