Function test_N

easygraph/classes/tests/test_hypergraph.py:640–646

(g1, g2)

def test_N(g1, g2):
    import torch

    assert (g1.N_v(0).cpu() == torch.tensor([0, 1, 2, 5])).all()
    assert (g1.N_e(2).cpu() == torch.tensor([0, 2])).all()
    assert (g2.N_v(1).cpu() == torch.tensor([0, 1, 3])).all()
    assert (g2.N_e(3).cpu() == torch.tensor([0, 1, 3, 4])).all()


def test_N_group(g1):