(self)
| 616 | [9, 1, 2, 9, 3]) |
| 617 | |
| 618 | def test_multidim(self): |
| 619 | a = [[1, 1, 1]] |
| 620 | r = [[2, 2, 2], |
| 621 | [1, 1, 1]] |
| 622 | assert_equal(insert(a, 0, [1]), [1, 1, 1, 1]) |
| 623 | assert_equal(insert(a, 0, [2, 2, 2], axis=0), r) |
| 624 | assert_equal(insert(a, 0, 2, axis=0), r) |
| 625 | assert_equal(insert(a, 2, 2, axis=1), [[1, 1, 2, 1]]) |
| 626 | |
| 627 | a = np.array([[1, 1], [2, 2], [3, 3]]) |
| 628 | b = np.arange(1, 4).repeat(3).reshape(3, 3) |
| 629 | c = np.concatenate( |
| 630 | (a[:, 0:1], np.arange(1, 4).repeat(3).reshape(3, 3).T, |
| 631 | a[:, 1:2]), axis=1) |
| 632 | assert_equal(insert(a, [1], [[1], [2], [3]], axis=1), b) |
| 633 | assert_equal(insert(a, [1], [1, 2, 3], axis=1), c) |
| 634 | # scalars behave differently, in this case exactly opposite: |
| 635 | assert_equal(insert(a, 1, [1, 2, 3], axis=1), b) |
| 636 | assert_equal(insert(a, 1, [[1], [2], [3]], axis=1), c) |
| 637 | |
| 638 | a = np.arange(4).reshape(2, 2) |
| 639 | assert_equal(insert(a[:, :1], 1, a[:, 1], axis=1), a) |
| 640 | assert_equal(insert(a[:1, :], 1, a[1, :], axis=0), a) |
| 641 | |
| 642 | # negative axis value |
| 643 | a = np.arange(24).reshape((2, 3, 4)) |
| 644 | assert_equal(insert(a, 1, a[:, :, 3], axis=-1), |
| 645 | insert(a, 1, a[:, :, 3], axis=2)) |
| 646 | assert_equal(insert(a, 1, a[:, 2, :], axis=-2), |
| 647 | insert(a, 1, a[:, 2, :], axis=1)) |
| 648 | |
| 649 | # invalid axis value |
| 650 | assert_raises(AxisError, insert, a, 1, a[:, 2, :], axis=3) |
| 651 | assert_raises(AxisError, insert, a, 1, a[:, 2, :], axis=-4) |
| 652 | |
| 653 | # negative axis value |
| 654 | a = np.arange(24).reshape((2, 3, 4)) |
| 655 | assert_equal(insert(a, 1, a[:, :, 3], axis=-1), |
| 656 | insert(a, 1, a[:, :, 3], axis=2)) |
| 657 | assert_equal(insert(a, 1, a[:, 2, :], axis=-2), |
| 658 | insert(a, 1, a[:, 2, :], axis=1)) |
| 659 | |
| 660 | def test_0d(self): |
| 661 | a = np.array(1) |
nothing calls this directly
no test coverage detected