| 3326 | return x, X, XX, m, mx, mX, mXX, m2x, m2X, m2XX |
| 3327 | |
| 3328 | def test_generic_methods(self): |
| 3329 | # Tests some MaskedArray methods. |
| 3330 | a = array([1, 3, 2]) |
| 3331 | assert_equal(a.any(), a._data.any()) |
| 3332 | assert_equal(a.all(), a._data.all()) |
| 3333 | assert_equal(a.argmax(), a._data.argmax()) |
| 3334 | assert_equal(a.argmin(), a._data.argmin()) |
| 3335 | assert_equal(a.choose(0, 1, 2, 3, 4), a._data.choose(0, 1, 2, 3, 4)) |
| 3336 | assert_equal(a.compress([1, 0, 1]), a._data.compress([1, 0, 1])) |
| 3337 | assert_equal(a.conj(), a._data.conj()) |
| 3338 | assert_equal(a.conjugate(), a._data.conjugate()) |
| 3339 | |
| 3340 | m = array([[1, 2], [3, 4]]) |
| 3341 | assert_equal(m.diagonal(), m._data.diagonal()) |
| 3342 | assert_equal(a.sum(), a._data.sum()) |
| 3343 | assert_equal(a.take([1, 2]), a._data.take([1, 2])) |
| 3344 | assert_equal(m.transpose(), m._data.transpose()) |
| 3345 | |
| 3346 | def test_allclose(self): |
| 3347 | # Tests allclose on arrays |