(self)
| 512 | assert_(w[0].category is FutureWarning) |
| 513 | |
| 514 | def test_multidim(self): |
| 515 | a = [[1, 1, 1]] |
| 516 | r = [[2, 2, 2], |
| 517 | [1, 1, 1]] |
| 518 | assert_equal(insert(a, 0, [1]), [1, 1, 1, 1]) |
| 519 | assert_equal(insert(a, 0, [2, 2, 2], axis=0), r) |
| 520 | assert_equal(insert(a, 0, 2, axis=0), r) |
| 521 | assert_equal(insert(a, 2, 2, axis=1), [[1, 1, 2, 1]]) |
| 522 | |
| 523 | a = np.array([[1, 1], [2, 2], [3, 3]]) |
| 524 | b = np.arange(1, 4).repeat(3).reshape(3, 3) |
| 525 | c = np.concatenate( |
| 526 | (a[:, 0:1], np.arange(1, 4).repeat(3).reshape(3, 3).T, |
| 527 | a[:, 1:2]), axis=1) |
| 528 | assert_equal(insert(a, [1], [[1], [2], [3]], axis=1), b) |
| 529 | assert_equal(insert(a, [1], [1, 2, 3], axis=1), c) |
| 530 | # scalars behave differently, in this case exactly opposite: |
| 531 | assert_equal(insert(a, 1, [1, 2, 3], axis=1), b) |
| 532 | assert_equal(insert(a, 1, [[1], [2], [3]], axis=1), c) |
| 533 | |
| 534 | a = np.arange(4).reshape(2, 2) |
| 535 | assert_equal(insert(a[:, :1], 1, a[:, 1], axis=1), a) |
| 536 | assert_equal(insert(a[:1,:], 1, a[1,:], axis=0), a) |
| 537 | |
| 538 | # negative axis value |
| 539 | a = np.arange(24).reshape((2, 3, 4)) |
| 540 | assert_equal(insert(a, 1, a[:,:, 3], axis=-1), |
| 541 | insert(a, 1, a[:,:, 3], axis=2)) |
| 542 | assert_equal(insert(a, 1, a[:, 2,:], axis=-2), |
| 543 | insert(a, 1, a[:, 2,:], axis=1)) |
| 544 | |
| 545 | # invalid axis value |
| 546 | assert_raises(np.AxisError, insert, a, 1, a[:, 2, :], axis=3) |
| 547 | assert_raises(np.AxisError, insert, a, 1, a[:, 2, :], axis=-4) |
| 548 | |
| 549 | # negative axis value |
| 550 | a = np.arange(24).reshape((2, 3, 4)) |
| 551 | assert_equal(insert(a, 1, a[:, :, 3], axis=-1), |
| 552 | insert(a, 1, a[:, :, 3], axis=2)) |
| 553 | assert_equal(insert(a, 1, a[:, 2, :], axis=-2), |
| 554 | insert(a, 1, a[:, 2, :], axis=1)) |
| 555 | |
| 556 | def test_0d(self): |
| 557 | a = np.array(1) |
nothing calls this directly
no test coverage detected