def test_canonical(): coords = np.array([[0, 0, 0], [0, 1, 0], [1, 0, 3], [0, 1, 0], [1, 0, 3]]).T data = np.arange(5) + 1 old = COO(coords, data, shape=(2, 2, 5)) x = COO(coords, data, shape=(2, 2, 5)) x.sum_duplicates() assert_eq(old, x) # assert x.nnz == 5 # assert x.has_duplicates assert x.nnz == 3 assert not x.has_duplicates
def test_canonical(): coords = np.array([[0, 0, 0], [0, 1, 0], [1, 0, 3], [0, 1, 0], [1, 0, 3]]).T data = np.arange(5) x = COO(coords, data, shape=(2, 2, 5)) y = x.sum_duplicates() assert_eq(x, y) assert x.nnz == 5 assert x.has_duplicates assert y.nnz == 3 assert not y.has_duplicates