예제 #1
0
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
예제 #2
0
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