()
| 1025 | |
| 1026 | |
| 1027 | def test_diagonal_zero_chunks(): |
| 1028 | x = da.ones((8, 8), chunks=(4, 4)) |
| 1029 | dd = da.ones((8, 8), chunks=(4, 4)) |
| 1030 | d = da.diagonal(dd) |
| 1031 | |
| 1032 | expected = np.ones((8,)) |
| 1033 | assert_eq(d, expected) |
| 1034 | assert_eq(d + d, 2 * expected) |
| 1035 | A = d + x |
| 1036 | assert_eq(A, np.full((8, 8), 2.0)) |
| 1037 | |
| 1038 | |
| 1039 | @pytest.mark.parametrize("fn", ["zeros_like", "ones_like"]) |