MCPcopy Create free account
hub / github.com/numpy/numpy / test_nan_behavior

Method test_nan_behavior

numpy/ma/tests/test_extras.py:1111–1144  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

1109 assert_equal(np.ma.median(d, out=o), o)
1110
1111 def test_nan_behavior(self):
1112 a = np.ma.masked_array(np.arange(24, dtype=float))
1113 a[::3] = np.ma.masked
1114 a[2] = np.nan
1115 assert_array_equal(np.ma.median(a), np.nan)
1116 assert_array_equal(np.ma.median(a, axis=0), np.nan)
1117
1118 a = np.ma.masked_array(np.arange(24, dtype=float).reshape(2, 3, 4))
1119 a.mask = np.arange(a.size) % 2 == 1
1120 aorig = a.copy()
1121 a[1, 2, 3] = np.nan
1122 a[1, 1, 2] = np.nan
1123
1124 # no axis
1125 assert_array_equal(np.ma.median(a), np.nan)
1126 assert_(np.isscalar(np.ma.median(a)))
1127
1128 # axis0
1129 b = np.ma.median(aorig, axis=0)
1130 b[2, 3] = np.nan
1131 b[1, 2] = np.nan
1132 assert_equal(np.ma.median(a, 0), b)
1133
1134 # axis1
1135 b = np.ma.median(aorig, axis=1)
1136 b[1, 3] = np.nan
1137 b[1, 2] = np.nan
1138 assert_equal(np.ma.median(a, 1), b)
1139
1140 # axis02
1141 b = np.ma.median(aorig, axis=(0, 2))
1142 b[1] = np.nan
1143 b[2] = np.nan
1144 assert_equal(np.ma.median(a, (0, 2)), b)
1145
1146 def test_ambigous_fill(self):
1147 # 255 is max value, used as filler for sort

Callers

nothing calls this directly

Calls 5

assert_array_equalFunction · 0.90
assert_equalFunction · 0.90
reshapeMethod · 0.80
assert_Function · 0.50
copyMethod · 0.45

Tested by

no test coverage detected