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())
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)
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)