| 739 | |
| 740 | |
| 741 | def test_trace(): |
| 742 | def _assert(a, b, *args, **kwargs): |
| 743 | return assert_eq(a.trace(*args, **kwargs), b.trace(*args, **kwargs)) |
| 744 | |
| 745 | b = np.arange(12).reshape((3, 4)) |
| 746 | a = da.from_array(b, 1) |
| 747 | _assert(a, b) |
| 748 | _assert(a, b, 0) |
| 749 | _assert(a, b, 1) |
| 750 | _assert(a, b, -1) |
| 751 | |
| 752 | b = np.arange(8).reshape((2, 2, 2)) |
| 753 | a = da.from_array(b, 2) |
| 754 | _assert(a, b) |
| 755 | _assert(a, b, 0) |
| 756 | _assert(a, b, 1) |
| 757 | _assert(a, b, -1) |
| 758 | _assert(a, b, 0, 0, 1) |
| 759 | _assert(a, b, 0, 0, 2) |
| 760 | _assert(a, b, 0, 1, 2, int) |
| 761 | _assert(a, b, 0, 1, 2, float) |
| 762 | _assert(a, b, offset=1, axis1=0, axis2=2, dtype=int) |
| 763 | _assert(a, b, offset=1, axis1=0, axis2=2, dtype=float) |
| 764 | |
| 765 | |
| 766 | @pytest.mark.parametrize("func", ["median", "nanmedian"]) |