| 555 | |
| 556 | @pytest.mark.parametrize("n", [0, 1, 2]) |
| 557 | def test_diff_prepend(n): |
| 558 | x = np.arange(5) + 1 |
| 559 | a = da.from_array(x, chunks=2) |
| 560 | assert_eq(da.diff(a, n, prepend=0), np.diff(x, n, prepend=0)) |
| 561 | assert_eq(da.diff(a, n, prepend=[0]), np.diff(x, n, prepend=[0])) |
| 562 | assert_eq(da.diff(a, n, prepend=[-1, 0]), np.diff(x, n, prepend=[-1, 0])) |
| 563 | |
| 564 | x = np.arange(16).reshape(4, 4) |
| 565 | a = da.from_array(x, chunks=2) |
| 566 | assert_eq(da.diff(a, n, axis=1, prepend=0), np.diff(x, n, axis=1, prepend=0)) |
| 567 | assert_eq( |
| 568 | da.diff(a, n, axis=1, prepend=[[0], [0], [0], [0]]), |
| 569 | np.diff(x, n, axis=1, prepend=[[0], [0], [0], [0]]), |
| 570 | ) |
| 571 | assert_eq(da.diff(a, n, axis=0, prepend=0), np.diff(x, n, axis=0, prepend=0)) |
| 572 | assert_eq( |
| 573 | da.diff(a, n, axis=0, prepend=[[0, 0, 0, 0]]), |
| 574 | np.diff(x, n, axis=0, prepend=[[0, 0, 0, 0]]), |
| 575 | ) |
| 576 | |
| 577 | if n > 0: |
| 578 | # When order is 0 the result is the input array, it doesn't raise |
| 579 | # an error |
| 580 | with pytest.raises(ValueError): |
| 581 | da.diff(a, n, prepend=np.zeros((3, 3))) |
| 582 | |
| 583 | |
| 584 | @pytest.mark.parametrize("n", [0, 1, 2]) |