(self)
| 583 | assert_(eq(wts, [1., 0.])) |
| 584 | |
| 585 | def test_testAverage2(self): |
| 586 | # More tests of average. |
| 587 | w1 = [0, 1, 1, 1, 1, 0] |
| 588 | w2 = [[0, 1, 1, 1, 1, 0], [1, 0, 0, 0, 0, 1]] |
| 589 | x = arange(6) |
| 590 | assert_(allclose(average(x, axis=0), 2.5)) |
| 591 | assert_(allclose(average(x, axis=0, weights=w1), 2.5)) |
| 592 | y = array([arange(6), 2.0 * arange(6)]) |
| 593 | assert_(allclose(average(y, None), |
| 594 | np.add.reduce(np.arange(6)) * 3. / 12.)) |
| 595 | assert_(allclose(average(y, axis=0), np.arange(6) * 3. / 2.)) |
| 596 | assert_(allclose(average(y, axis=1), |
| 597 | [average(x, axis=0), average(x, axis=0)*2.0])) |
| 598 | assert_(allclose(average(y, None, weights=w2), 20. / 6.)) |
| 599 | assert_(allclose(average(y, axis=0, weights=w2), |
| 600 | [0., 1., 2., 3., 4., 10.])) |
| 601 | assert_(allclose(average(y, axis=1), |
| 602 | [average(x, axis=0), average(x, axis=0)*2.0])) |
| 603 | m1 = zeros(6) |
| 604 | m2 = [0, 0, 1, 1, 0, 0] |
| 605 | m3 = [[0, 0, 1, 1, 0, 0], [0, 1, 1, 1, 1, 0]] |
| 606 | m4 = ones(6) |
| 607 | m5 = [0, 1, 1, 1, 1, 1] |
| 608 | assert_(allclose(average(masked_array(x, m1), axis=0), 2.5)) |
| 609 | assert_(allclose(average(masked_array(x, m2), axis=0), 2.5)) |
| 610 | assert_(average(masked_array(x, m4), axis=0) is masked) |
| 611 | assert_equal(average(masked_array(x, m5), axis=0), 0.0) |
| 612 | assert_equal(count(average(masked_array(x, m4), axis=0)), 0) |
| 613 | z = masked_array(y, m3) |
| 614 | assert_(allclose(average(z, None), 20. / 6.)) |
| 615 | assert_(allclose(average(z, axis=0), |
| 616 | [0., 1., 99., 99., 4.0, 7.5])) |
| 617 | assert_(allclose(average(z, axis=1), [2.5, 5.0])) |
| 618 | assert_(allclose(average(z, axis=0, weights=w2), |
| 619 | [0., 1., 99., 99., 4.0, 10.0])) |
| 620 | |
| 621 | a = arange(6) |
| 622 | b = arange(6) * 3 |
| 623 | r1, w1 = average([[a, b], [b, a]], axis=1, returned=True) |
| 624 | assert_equal(shape(r1), shape(w1)) |
| 625 | assert_equal(r1.shape, w1.shape) |
| 626 | r2, w2 = average(ones((2, 2, 3)), axis=0, weights=[3, 1], returned=True) |
| 627 | assert_equal(shape(w2), shape(r2)) |
| 628 | r2, w2 = average(ones((2, 2, 3)), returned=True) |
| 629 | assert_equal(shape(w2), shape(r2)) |
| 630 | r2, w2 = average(ones((2, 2, 3)), weights=ones((2, 2, 3)), returned=True) |
| 631 | assert_(shape(w2) == shape(r2)) |
| 632 | a2d = array([[1, 2], [0, 4]], float) |
| 633 | a2dm = masked_array(a2d, [[0, 0], [1, 0]]) |
| 634 | a2da = average(a2d, axis=0) |
| 635 | assert_(eq(a2da, [0.5, 3.0])) |
| 636 | a2dma = average(a2dm, axis=0) |
| 637 | assert_(eq(a2dma, [1.0, 3.0])) |
| 638 | a2dma = average(a2dm, axis=None) |
| 639 | assert_(eq(a2dma, 7. / 3.)) |
| 640 | a2dma = average(a2dm, axis=1) |
| 641 | assert_(eq(a2dma, [1.5, 4.0])) |
| 642 |
nothing calls this directly
no test coverage detected