(self)
| 1069 | expected) |
| 1070 | |
| 1071 | def test_nan(self): |
| 1072 | for mask in (False, np.zeros(6, dtype=bool)): |
| 1073 | dm = np.ma.array([[1, np.nan, 3], [1, 2, 3]]) |
| 1074 | dm.mask = mask |
| 1075 | |
| 1076 | # scalar result |
| 1077 | r = np.ma.median(dm, axis=None) |
| 1078 | assert_(np.isscalar(r)) |
| 1079 | assert_array_equal(r, np.nan) |
| 1080 | r = np.ma.median(dm.ravel(), axis=0) |
| 1081 | assert_(np.isscalar(r)) |
| 1082 | assert_array_equal(r, np.nan) |
| 1083 | |
| 1084 | r = np.ma.median(dm, axis=0) |
| 1085 | assert_equal(type(r), MaskedArray) |
| 1086 | assert_array_equal(r, [1, np.nan, 3]) |
| 1087 | r = np.ma.median(dm, axis=1) |
| 1088 | assert_equal(type(r), MaskedArray) |
| 1089 | assert_array_equal(r, [np.nan, 2]) |
| 1090 | r = np.ma.median(dm, axis=-1) |
| 1091 | assert_equal(type(r), MaskedArray) |
| 1092 | assert_array_equal(r, [np.nan, 2]) |
| 1093 | |
| 1094 | dm = np.ma.array([[1, np.nan, 3], [1, 2, 3]]) |
| 1095 | dm[:, 2] = np.ma.masked |
| 1096 | assert_array_equal(np.ma.median(dm, axis=None), np.nan) |
| 1097 | assert_array_equal(np.ma.median(dm, axis=0), [1, np.nan, 3]) |
| 1098 | assert_array_equal(np.ma.median(dm, axis=1), [np.nan, 1.5]) |
| 1099 | |
| 1100 | def test_out_nan(self): |
| 1101 | o = np.ma.masked_array(np.zeros((4,))) |
nothing calls this directly
no test coverage detected