diff options
author | Marten H. van Kerkwijk <mhvk@weasel.astro.utoronto.ca> | 2018-01-02 14:40:56 -0500 |
---|---|---|
committer | Marten van Kerkwijk <mhvk@astro.utoronto.ca> | 2018-05-16 11:01:33 -0400 |
commit | 6756edfad754ed1b65f44e8fe84cf2c9d2f53ce9 (patch) | |
tree | 638731f8a8c76d7a5de0cc4a09ce7c81ea1ca760 /numpy/ma/tests/test_regression.py | |
parent | 290aa88e50e65c2dd5df4ddb062b895e9e7baae3 (diff) | |
download | numpy-6756edfad754ed1b65f44e8fe84cf2c9d2f53ce9.tar.gz |
BUG: Ensure __array_finalize__ cannot back-mangle shape
Diffstat (limited to 'numpy/ma/tests/test_regression.py')
-rw-r--r-- | numpy/ma/tests/test_regression.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/numpy/ma/tests/test_regression.py b/numpy/ma/tests/test_regression.py index 04e10d9d1..96c418a51 100644 --- a/numpy/ma/tests/test_regression.py +++ b/numpy/ma/tests/test_regression.py @@ -74,3 +74,13 @@ class TestRegression(object): r1 = np.ma.corrcoef(x, y, ddof=1) # ddof should not have an effect (it gets cancelled out) assert_allclose(r0.data, r1.data) + + 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)) |