| 813 | |
| 814 | |
| 815 | def test_trace(): |
| 816 | def _assert(a, b, *args, **kwargs): |
| 817 | return assert_eq(a.trace(*args, **kwargs), b.trace(*args, **kwargs)) |
| 818 | |
| 819 | b = np.arange(12).reshape((3, 4)) |
| 820 | a = da.from_array(b, 1) |
| 821 | _assert(a, b) |
| 822 | _assert(a, b, 0) |
| 823 | _assert(a, b, 1) |
| 824 | _assert(a, b, -1) |
| 825 | |
| 826 | b = np.arange(8).reshape((2, 2, 2)) |
| 827 | a = da.from_array(b, 2) |
| 828 | _assert(a, b) |
| 829 | _assert(a, b, 0) |
| 830 | _assert(a, b, 1) |
| 831 | _assert(a, b, -1) |
| 832 | _assert(a, b, 0, 0, 1) |
| 833 | _assert(a, b, 0, 0, 2) |
| 834 | _assert(a, b, 0, 1, 2, int) |
| 835 | _assert(a, b, 0, 1, 2, float) |
| 836 | _assert(a, b, offset=1, axis1=0, axis2=2, dtype=int) |
| 837 | _assert(a, b, offset=1, axis1=0, axis2=2, dtype=float) |
| 838 | |
| 839 | |
| 840 | @pytest.mark.parametrize("func", ["median", "nanmedian"]) |