Function create_test_graph

tests/python/common/test_heterograph-shared-memory.py:18–30

(idtype)

def create_test_graph(idtype):
    g = dgl.heterograph(
        (
            {
                ("user", "follows", "user"): ([0, 1], [1, 2]),
                ("user", "plays", "game"): ([0, 1, 2, 1], [0, 0, 1, 1]),
                ("user", "wishes", "game"): ([0, 2], [1, 0]),
                ("developer", "develops", "game"): ([0, 1], [0, 1]),
            }
        ),
        idtype=idtype,
    )
    return g


def _assert_is_identical_hetero(g, g2):