(self)
| 376 | |
| 377 | @np.errstate(all='ignore') |
| 378 | def test_99(self): |
| 379 | # test average |
| 380 | ott = self.array([0., 1., 2., 3.], mask=[1, 0, 0, 0]) |
| 381 | self.assert_array_equal(2.0, self.average(ott, axis=0)) |
| 382 | self.assert_array_equal(2.0, self.average(ott, weights=[1., 1., 2., 1.])) |
| 383 | result, wts = self.average(ott, weights=[1., 1., 2., 1.], returned=1) |
| 384 | self.assert_array_equal(2.0, result) |
| 385 | assert(wts == 4.0) |
| 386 | ott[:] = self.masked |
| 387 | assert(self.average(ott, axis=0) is self.masked) |
| 388 | ott = self.array([0., 1., 2., 3.], mask=[1, 0, 0, 0]) |
| 389 | ott = ott.reshape(2, 2) |
| 390 | ott[:, 1] = self.masked |
| 391 | self.assert_array_equal(self.average(ott, axis=0), [2.0, 0.0]) |
| 392 | assert(self.average(ott, axis=1)[0] is self.masked) |
| 393 | self.assert_array_equal([2., 0.], self.average(ott, axis=0)) |
| 394 | result, wts = self.average(ott, axis=0, returned=1) |
| 395 | self.assert_array_equal(wts, [1., 0.]) |
| 396 | w1 = [0, 1, 1, 1, 1, 0] |
| 397 | w2 = [[0, 1, 1, 1, 1, 0], [1, 0, 0, 0, 0, 1]] |
| 398 | x = self.arange(6) |
| 399 | self.assert_array_equal(self.average(x, axis=0), 2.5) |
| 400 | self.assert_array_equal(self.average(x, axis=0, weights=w1), 2.5) |
| 401 | y = self.array([self.arange(6), 2.0*self.arange(6)]) |
| 402 | self.assert_array_equal(self.average(y, None), np.add.reduce(np.arange(6))*3./12.) |
| 403 | self.assert_array_equal(self.average(y, axis=0), np.arange(6) * 3./2.) |
| 404 | self.assert_array_equal(self.average(y, axis=1), [self.average(x, axis=0), self.average(x, axis=0) * 2.0]) |
| 405 | self.assert_array_equal(self.average(y, None, weights=w2), 20./6.) |
| 406 | self.assert_array_equal(self.average(y, axis=0, weights=w2), [0., 1., 2., 3., 4., 10.]) |
| 407 | self.assert_array_equal(self.average(y, axis=1), [self.average(x, axis=0), self.average(x, axis=0) * 2.0]) |
| 408 | m1 = self.zeros(6) |
| 409 | m2 = [0, 0, 1, 1, 0, 0] |
| 410 | m3 = [[0, 0, 1, 1, 0, 0], [0, 1, 1, 1, 1, 0]] |
| 411 | m4 = self.ones(6) |
| 412 | m5 = [0, 1, 1, 1, 1, 1] |
| 413 | self.assert_array_equal(self.average(self.masked_array(x, m1), axis=0), 2.5) |
| 414 | self.assert_array_equal(self.average(self.masked_array(x, m2), axis=0), 2.5) |
| 415 | self.assert_array_equal(self.average(self.masked_array(x, m5), axis=0), 0.0) |
| 416 | self.assert_array_equal(self.count(self.average(self.masked_array(x, m4), axis=0)), 0) |
| 417 | z = self.masked_array(y, m3) |
| 418 | self.assert_array_equal(self.average(z, None), 20./6.) |
| 419 | self.assert_array_equal(self.average(z, axis=0), [0., 1., 99., 99., 4.0, 7.5]) |
| 420 | self.assert_array_equal(self.average(z, axis=1), [2.5, 5.0]) |
| 421 | self.assert_array_equal(self.average(z, axis=0, weights=w2), [0., 1., 99., 99., 4.0, 10.0]) |
| 422 | |
| 423 | @np.errstate(all='ignore') |
| 424 | def test_A(self): |
nothing calls this directly
no test coverage detected