| 219 | [0., 1., 99., 99., 4.0, 10.0]) |
| 220 | |
| 221 | def test_testAverage3(self): |
| 222 | # Yet more tests of average! |
| 223 | a = arange(6) |
| 224 | b = arange(6) * 3 |
| 225 | r1, w1 = average([[a, b], [b, a]], axis=1, returned=True) |
| 226 | assert_equal(shape(r1), shape(w1)) |
| 227 | assert_equal(r1.shape, w1.shape) |
| 228 | r2, w2 = average(ones((2, 2, 3)), axis=0, weights=[3, 1], returned=True) |
| 229 | assert_equal(shape(w2), shape(r2)) |
| 230 | r2, w2 = average(ones((2, 2, 3)), returned=True) |
| 231 | assert_equal(shape(w2), shape(r2)) |
| 232 | r2, w2 = average(ones((2, 2, 3)), weights=ones((2, 2, 3)), returned=True) |
| 233 | assert_equal(shape(w2), shape(r2)) |
| 234 | a2d = array([[1, 2], [0, 4]], float) |
| 235 | a2dm = masked_array(a2d, [[False, False], [True, False]]) |
| 236 | a2da = average(a2d, axis=0) |
| 237 | assert_equal(a2da, [0.5, 3.0]) |
| 238 | a2dma = average(a2dm, axis=0) |
| 239 | assert_equal(a2dma, [1.0, 3.0]) |
| 240 | a2dma = average(a2dm, axis=None) |
| 241 | assert_equal(a2dma, 7. / 3.) |
| 242 | a2dma = average(a2dm, axis=1) |
| 243 | assert_equal(a2dma, [1.5, 4.0]) |
| 244 | |
| 245 | def test_testAverage4(self): |
| 246 | # Test that `keepdims` works with average |