Exemplo n.º 1
0
def test_add_self_loops_sparse():
    source = csc_matrix(test_matrices[3][0])
    target = np.matrix(test_matrices[3][1])
    
    looped = mc.add_self_loops(source, 2).todense()
    assert np.array_equal(looped, target)

    lv = 1.2940192

    for _ in range(100):
        x = np.random.randint(0, 2, size=(5, 5)) * np.random.rand(5, 5)
        x_cs = csc_matrix(x)

        np.fill_diagonal(x, lv)

        assert np.allclose(x, mc.add_self_loops(x_cs, lv).todense())
Exemplo n.º 2
0
def test_add_self_loops_sparse():
    source = csc_matrix(test_matrices[3][0])
    target = np.matrix(test_matrices[3][1])

    looped = mc.add_self_loops(source, 2).todense()
    assert np.array_equal(looped, target)
Exemplo n.º 3
0
def test_add_self_loops():
    source = np.matrix(test_matrices[3][0])
    target = np.matrix(test_matrices[3][1])

    looped = mc.add_self_loops(source, 2)
    assert np.array_equal(looped, target)