(self)
| 645 | assert_(eq(wts, [1., 0.])) |
| 646 | |
| 647 | def test_testAverage2(self): |
| 648 | # More tests of average. |
| 649 | w1 = [0, 1, 1, 1, 1, 0] |
| 650 | w2 = [[0, 1, 1, 1, 1, 0], [1, 0, 0, 0, 0, 1]] |
| 651 | x = arange(6) |
| 652 | assert_(allclose(average(x, axis=0), 2.5)) |
| 653 | assert_(allclose(average(x, axis=0, weights=w1), 2.5)) |
| 654 | y = array([arange(6), 2.0 * arange(6)]) |
| 655 | assert_(allclose(average(y, None), |
| 656 | np.add.reduce(np.arange(6)) * 3. / 12.)) |
| 657 | assert_(allclose(average(y, axis=0), np.arange(6) * 3. / 2.)) |
| 658 | assert_(allclose(average(y, axis=1), |
| 659 | [average(x, axis=0), average(x, axis=0) * 2.0])) |
| 660 | assert_(allclose(average(y, None, weights=w2), 20. / 6.)) |
| 661 | assert_(allclose(average(y, axis=0, weights=w2), |
| 662 | [0., 1., 2., 3., 4., 10.])) |
| 663 | assert_(allclose(average(y, axis=1), |
| 664 | [average(x, axis=0), average(x, axis=0) * 2.0])) |
| 665 | m1 = zeros(6) |
| 666 | m2 = [0, 0, 1, 1, 0, 0] |
| 667 | m3 = [[0, 0, 1, 1, 0, 0], [0, 1, 1, 1, 1, 0]] |
| 668 | m4 = ones(6) |
| 669 | m5 = [0, 1, 1, 1, 1, 1] |
| 670 | assert_(allclose(average(masked_array(x, m1), axis=0), 2.5)) |
| 671 | assert_(allclose(average(masked_array(x, m2), axis=0), 2.5)) |
| 672 | assert_(average(masked_array(x, m4), axis=0) is masked) |
| 673 | assert_equal(average(masked_array(x, m5), axis=0), 0.0) |
| 674 | assert_equal(count(average(masked_array(x, m4), axis=0)), 0) |
| 675 | z = masked_array(y, m3) |
| 676 | assert_(allclose(average(z, None), 20. / 6.)) |
| 677 | assert_(allclose(average(z, axis=0), |
| 678 | [0., 1., 99., 99., 4.0, 7.5])) |
| 679 | assert_(allclose(average(z, axis=1), [2.5, 5.0])) |
| 680 | assert_(allclose(average(z, axis=0, weights=w2), |
| 681 | [0., 1., 99., 99., 4.0, 10.0])) |
| 682 | |
| 683 | a = arange(6) |
| 684 | b = arange(6) * 3 |
| 685 | r1, w1 = average([[a, b], [b, a]], axis=1, returned=True) |
| 686 | assert_equal(shape(r1), shape(w1)) |
| 687 | assert_equal(r1.shape, w1.shape) |
| 688 | r2, w2 = average(ones((2, 2, 3)), axis=0, weights=[3, 1], returned=True) |
| 689 | assert_equal(shape(w2), shape(r2)) |
| 690 | r2, w2 = average(ones((2, 2, 3)), returned=True) |
| 691 | assert_equal(shape(w2), shape(r2)) |
| 692 | r2, w2 = average(ones((2, 2, 3)), weights=ones((2, 2, 3)), returned=True) |
| 693 | assert_(shape(w2) == shape(r2)) |
| 694 | a2d = array([[1, 2], [0, 4]], float) |
| 695 | a2dm = masked_array(a2d, [[0, 0], [1, 0]]) |
| 696 | a2da = average(a2d, axis=0) |
| 697 | assert_(eq(a2da, [0.5, 3.0])) |
| 698 | a2dma = average(a2dm, axis=0) |
| 699 | assert_(eq(a2dma, [1.0, 3.0])) |
| 700 | a2dma = average(a2dm, axis=None) |
| 701 | assert_(eq(a2dma, 7. / 3.)) |
| 702 | a2dma = average(a2dm, axis=1) |
| 703 | assert_(eq(a2dma, [1.5, 4.0])) |
| 704 |
nothing calls this directly
no test coverage detected