def test_sparse_eye(): res = sparse_eye(5) actual = torch.eye(5) assert torch.equal(res.to_dense(), actual)