diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2017-12-18 22:53:17 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-12-18 22:53:17 -0700 |
| commit | 0225771c13e2cf54dc3dd1ff4d39af6a78fde9d8 (patch) | |
| tree | bcccc559e08f8ac187944fb4890a5f4ea132af55 /numpy/ma/tests | |
| parent | 0f13a2e7fa626dbde62bdda7a92fc0a6361981ba (diff) | |
| parent | 4c8376223c717ff63548f7fad65975f0a88eda41 (diff) | |
| download | numpy-0225771c13e2cf54dc3dd1ff4d39af6a78fde9d8.tar.gz | |
Merge pull request #10232 from eric-wieser/use-isclose-in-masked_values
BUG: Don't reimplement isclose in np.ma
Diffstat (limited to 'numpy/ma/tests')
| -rw-r--r-- | numpy/ma/tests/test_core.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py index 2a5a65e5c..d5622e4bb 100644 --- a/numpy/ma/tests/test_core.py +++ b/numpy/ma/tests/test_core.py @@ -4982,6 +4982,21 @@ class TestMaskedConstant(object): assert_not_equal(repr(a), 'masked') +class TestMaskedWhereAliases(object): + + # TODO: Test masked_object, masked_equal, ... + + def test_masked_values(self): + res = masked_values(np.array([-32768.0]), np.int16(-32768)) + assert_equal(res.mask, [True]) + + res = masked_values(np.inf, np.inf) + assert_equal(res.mask, True) + + res = np.ma.masked_values(np.inf, -np.inf) + assert_equal(res.mask, False) + + def test_masked_array(): a = np.ma.array([0, 1, 2, 3], mask=[0, 0, 1, 0]) assert_equal(np.argwhere(a), [[1], [3]]) |