(self)
| 574 | class TestCompressFunctions: |
| 575 | |
| 576 | def test_compress_nd(self): |
| 577 | # Tests compress_nd |
| 578 | x = np.array(list(range(3 * 4 * 5))).reshape(3, 4, 5) |
| 579 | m = np.zeros((3, 4, 5)).astype(bool) |
| 580 | m[1, 1, 1] = True |
| 581 | x = array(x, mask=m) |
| 582 | |
| 583 | # axis=None |
| 584 | a = compress_nd(x) |
| 585 | assert_equal(a, [[[ 0, 2, 3, 4], |
| 586 | [10, 12, 13, 14], |
| 587 | [15, 17, 18, 19]], |
| 588 | [[40, 42, 43, 44], |
| 589 | [50, 52, 53, 54], |
| 590 | [55, 57, 58, 59]]]) |
| 591 | |
| 592 | # axis=0 |
| 593 | a = compress_nd(x, 0) |
| 594 | assert_equal(a, [[[ 0, 1, 2, 3, 4], |
| 595 | [ 5, 6, 7, 8, 9], |
| 596 | [10, 11, 12, 13, 14], |
| 597 | [15, 16, 17, 18, 19]], |
| 598 | [[40, 41, 42, 43, 44], |
| 599 | [45, 46, 47, 48, 49], |
| 600 | [50, 51, 52, 53, 54], |
| 601 | [55, 56, 57, 58, 59]]]) |
| 602 | |
| 603 | # axis=1 |
| 604 | a = compress_nd(x, 1) |
| 605 | assert_equal(a, [[[ 0, 1, 2, 3, 4], |
| 606 | [10, 11, 12, 13, 14], |
| 607 | [15, 16, 17, 18, 19]], |
| 608 | [[20, 21, 22, 23, 24], |
| 609 | [30, 31, 32, 33, 34], |
| 610 | [35, 36, 37, 38, 39]], |
| 611 | [[40, 41, 42, 43, 44], |
| 612 | [50, 51, 52, 53, 54], |
| 613 | [55, 56, 57, 58, 59]]]) |
| 614 | |
| 615 | a2 = compress_nd(x, (1,)) |
| 616 | a3 = compress_nd(x, -2) |
| 617 | a4 = compress_nd(x, (-2,)) |
| 618 | assert_equal(a, a2) |
| 619 | assert_equal(a, a3) |
| 620 | assert_equal(a, a4) |
| 621 | |
| 622 | # axis=2 |
| 623 | a = compress_nd(x, 2) |
| 624 | assert_equal(a, [[[ 0, 2, 3, 4], |
| 625 | [ 5, 7, 8, 9], |
| 626 | [10, 12, 13, 14], |
| 627 | [15, 17, 18, 19]], |
| 628 | [[20, 22, 23, 24], |
| 629 | [25, 27, 28, 29], |
| 630 | [30, 32, 33, 34], |
| 631 | [35, 37, 38, 39]], |
| 632 | [[40, 42, 43, 44], |
| 633 | [45, 47, 48, 49], |
nothing calls this directly
no test coverage detected