diff options
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/ma/tests/test_core.py | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py index f907a3447..5a8262e0d 100644 --- a/numpy/ma/tests/test_core.py +++ b/numpy/ma/tests/test_core.py @@ -101,8 +101,8 @@ class TestMaskedArray(TestCase): self.assertTrue(isMaskedArray(xm)) assert_equal(shape(xm), s) assert_equal(xm.shape, s) - assert_equal(xm.size , reduce(lambda x, y:x * y, s)) - assert_equal(count(xm) , len(m1) - reduce(lambda x, y:x + y, m1)) + assert_equal(xm.size, reduce(lambda x, y:x * y, s)) + assert_equal(count(xm), len(m1) - reduce(lambda x, y:x + y, m1)) assert_equal(xm, xf) assert_equal(filled(xm, 1.e20), xf) assert_equal(x, xm) @@ -1150,9 +1150,19 @@ class TestMaskedArrayArithmetic(TestCase): def test_eq_w_None(self): + # With no mask a = array([1, 2], mask=False) assert_equal(a == None, False) assert_equal(a != None, True) + # With a partial mask + a = array([1, 2], mask=[0, 1]) + assert_equal(a == None, False) + assert_equal(a != None, True) + # With total mask + a = array([1, 2], mask=True) + assert_equal(a == None, False) + assert_equal(a != None, True) + # As masked a = masked assert_equal(a == None, masked) |