(self)
| 486 | class TestCompressFunctions: |
| 487 | |
| 488 | def test_compress_nd(self): |
| 489 | # Tests compress_nd |
| 490 | x = np.array(list(range(3*4*5))).reshape(3, 4, 5) |
| 491 | m = np.zeros((3,4,5)).astype(bool) |
| 492 | m[1,1,1] = True |
| 493 | x = array(x, mask=m) |
| 494 | |
| 495 | # axis=None |
| 496 | a = compress_nd(x) |
| 497 | assert_equal(a, [[[ 0, 2, 3, 4], |
| 498 | [10, 12, 13, 14], |
| 499 | [15, 17, 18, 19]], |
| 500 | [[40, 42, 43, 44], |
| 501 | [50, 52, 53, 54], |
| 502 | [55, 57, 58, 59]]]) |
| 503 | |
| 504 | # axis=0 |
| 505 | a = compress_nd(x, 0) |
| 506 | assert_equal(a, [[[ 0, 1, 2, 3, 4], |
| 507 | [ 5, 6, 7, 8, 9], |
| 508 | [10, 11, 12, 13, 14], |
| 509 | [15, 16, 17, 18, 19]], |
| 510 | [[40, 41, 42, 43, 44], |
| 511 | [45, 46, 47, 48, 49], |
| 512 | [50, 51, 52, 53, 54], |
| 513 | [55, 56, 57, 58, 59]]]) |
| 514 | |
| 515 | # axis=1 |
| 516 | a = compress_nd(x, 1) |
| 517 | assert_equal(a, [[[ 0, 1, 2, 3, 4], |
| 518 | [10, 11, 12, 13, 14], |
| 519 | [15, 16, 17, 18, 19]], |
| 520 | [[20, 21, 22, 23, 24], |
| 521 | [30, 31, 32, 33, 34], |
| 522 | [35, 36, 37, 38, 39]], |
| 523 | [[40, 41, 42, 43, 44], |
| 524 | [50, 51, 52, 53, 54], |
| 525 | [55, 56, 57, 58, 59]]]) |
| 526 | |
| 527 | a2 = compress_nd(x, (1,)) |
| 528 | a3 = compress_nd(x, -2) |
| 529 | a4 = compress_nd(x, (-2,)) |
| 530 | assert_equal(a, a2) |
| 531 | assert_equal(a, a3) |
| 532 | assert_equal(a, a4) |
| 533 | |
| 534 | # axis=2 |
| 535 | a = compress_nd(x, 2) |
| 536 | assert_equal(a, [[[ 0, 2, 3, 4], |
| 537 | [ 5, 7, 8, 9], |
| 538 | [10, 12, 13, 14], |
| 539 | [15, 17, 18, 19]], |
| 540 | [[20, 22, 23, 24], |
| 541 | [25, 27, 28, 29], |
| 542 | [30, 32, 33, 34], |
| 543 | [35, 37, 38, 39]], |
| 544 | [[40, 42, 43, 44], |
| 545 | [45, 47, 48, 49], |
nothing calls this directly
no test coverage detected