Function test_L_sym_group

easygraph/classes/tests/test_hypergraph.py:747–779

(g1)

    reason="python requires >= 3.7",
)
def test_L_sym_group(g1):
    import torch

    g1.add_hyperedges([[0, 1]], group_name="knn")
    # all
    H = g1.H.to_dense().cpu()
    D_v_neg_1_2 = torch.diag(H.sum(dim=1).view(-1) ** (-0.5))
    D_e_neg_1 = torch.diag(H.sum(dim=0).view(-1) ** (-1))
    W_e = g1.W_e.to_dense()
    L_sym = (
        torch.eye(H.shape[0]) - D_v_neg_1_2 @ H @ W_e @ D_e_neg_1 @ H.t() @ D_v_neg_1_2
    )
    assert (L_sym == g1.L_sym.to_dense().cpu()).all()
    # main group
    H = g1.H_of_group("main").to_dense().cpu()
    D_v_neg_1_2 = torch.diag(H.sum(dim=1).view(-1) ** (-0.5))
    D_e_neg_1 = torch.diag(H.sum(dim=0).view(-1) ** (-1))
    W_e = g1.W_e_of_group("main").to_dense()
    L_sym = (
        torch.eye(H.shape[0]) - D_v_neg_1_2 @ H @ W_e @ D_e_neg_1 @ H.t() @ D_v_neg_1_2
    )
    assert (L_sym == g1.L_sym_of_group("main").to_dense().cpu()).all()
    # knn group
    H = g1.H_of_group("knn").to_dense().cpu()
    D_v_neg_1_2 = H.sum(dim=1).view(-1) ** (-0.5)
    D_v_neg_1_2[torch.isinf(D_v_neg_1_2)] = 0
    D_v_neg_1_2 = torch.diag(D_v_neg_1_2)
    D_e_neg_1 = torch.diag(H.sum(dim=0).view(-1) ** (-1))
    W_e = g1.W_e_of_group("knn").to_dense()
    L_sym = (
        torch.eye(H.shape[0]) - D_v_neg_1_2 @ H @ W_e @ D_e_neg_1 @ H.t() @ D_v_neg_1_2
    )
    assert (L_sym == g1.L_sym_of_group("knn").to_dense().cpu()).all()


@pytest.mark.skipif(