(self)
| 1157 | expected) |
| 1158 | |
| 1159 | def test_nan(self): |
| 1160 | for mask in (False, np.zeros(6, dtype=bool)): |
| 1161 | dm = np.ma.array([[1, np.nan, 3], [1, 2, 3]]) |
| 1162 | dm.mask = mask |
| 1163 | |
| 1164 | # scalar result |
| 1165 | r = np.ma.median(dm, axis=None) |
| 1166 | assert_(np.isscalar(r)) |
| 1167 | assert_array_equal(r, np.nan) |
| 1168 | r = np.ma.median(dm.ravel(), axis=0) |
| 1169 | assert_(np.isscalar(r)) |
| 1170 | assert_array_equal(r, np.nan) |
| 1171 | |
| 1172 | r = np.ma.median(dm, axis=0) |
| 1173 | assert_equal(type(r), MaskedArray) |
| 1174 | assert_array_equal(r, [1, np.nan, 3]) |
| 1175 | r = np.ma.median(dm, axis=1) |
| 1176 | assert_equal(type(r), MaskedArray) |
| 1177 | assert_array_equal(r, [np.nan, 2]) |
| 1178 | r = np.ma.median(dm, axis=-1) |
| 1179 | assert_equal(type(r), MaskedArray) |
| 1180 | assert_array_equal(r, [np.nan, 2]) |
| 1181 | |
| 1182 | dm = np.ma.array([[1, np.nan, 3], [1, 2, 3]]) |
| 1183 | dm[:, 2] = np.ma.masked |
| 1184 | assert_array_equal(np.ma.median(dm, axis=None), np.nan) |
| 1185 | assert_array_equal(np.ma.median(dm, axis=0), [1, np.nan, 3]) |
| 1186 | assert_array_equal(np.ma.median(dm, axis=1), [np.nan, 1.5]) |
| 1187 | |
| 1188 | def test_out_nan(self): |
| 1189 | o = np.ma.masked_array(np.zeros((4,))) |
nothing calls this directly
no test coverage detected