summaryrefslogtreecommitdiff
path: root/numpy
diff options
context:
space:
mode:
authorpierregm <pierregm@localhost>2009-02-07 18:51:31 +0000
committerpierregm <pierregm@localhost>2009-02-07 18:51:31 +0000
commitdab616fe622b7669278be1a139169bf323718241 (patch)
tree1ffdf50267f5fb12b877527c6c484fd50ca4c19d /numpy
parentc180e36cf4438947de5ba38c90ec30e9538f2f10 (diff)
downloadnumpy-dab616fe622b7669278be1a139169bf323718241.tar.gz
Diffstat (limited to 'numpy')
-rw-r--r--numpy/ma/tests/test_core.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py
index f13ff3377..c92c650b4 100644
--- a/numpy/ma/tests/test_core.py
+++ b/numpy/ma/tests/test_core.py
@@ -1124,12 +1124,12 @@ class TestMaskedArrayAttributes(TestCase):
def test_flat(self):
"Test flat on masked_matrices"
- test = ma.array(np.matrix([[1, 2, 3]]), mask=[0, 0, 1])
- test.flat = ma.array([3, 2, 1], mask=[1, 0, 0])
- control = ma.array(np.matrix([[3, 2, 1]]), mask=[1, 0, 0])
+ test = masked_array(np.matrix([[1, 2, 3]]), mask=[0, 0, 1])
+ test.flat = masked_array([3, 2, 1], mask=[1, 0, 0])
+ control = masked_array(np.matrix([[3, 2, 1]]), mask=[1, 0, 0])
assert_equal(test, control)
#
- test = ma.array(np.matrix([[1, 2, 3]]), mask=[0, 0, 1])
+ test = masked_array(np.matrix([[1, 2, 3]]), mask=[0, 0, 1])
testflat = test.flat
testflat[:] = testflat[[2, 1, 0]]
assert_equal(test, control)