diff options
Diffstat (limited to 'numpy/ma/tests')
| -rw-r--r-- | numpy/ma/tests/test_core.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py index 2dc758513..9b4f19284 100644 --- a/numpy/ma/tests/test_core.py +++ b/numpy/ma/tests/test_core.py @@ -1388,6 +1388,12 @@ class TestArrayMethods(NumpyTestCase): putmask(mxx, mask, values) assert_equal(mxx, [1,2,30,4,5,60]) + def test_compress(self): + a = array([1,2,3],mask=[True,False,False]) + b = compress(a,a<3) + assert_equal(b,[1,2]) + assert_equal(b.mask,[True,False]) + #.............................................................................. |