(A, row, col, val, shape)
| 16 | |
| 17 | |
| 18 | def all_close_sparse(A, row, col, val, shape): |
| 19 | rowA, colA = A.coo() |
| 20 | valA = A.val |
| 21 | assert torch.allclose(rowA, row) |
| 22 | assert torch.allclose(colA, col) |
| 23 | assert torch.allclose(valA, val) |
| 24 | assert A.shape == shape |
| 25 | |
| 26 | |
| 27 | @pytest.mark.parametrize( |