def test_qm7():
dataset = datasets.QM7()
dl = DisjointLoader(dataset, batch_size=batch_size)
dl.__next__()
bl = BatchLoader(dataset, batch_size=batch_size)
bl.__next__()
def test_qm9():
dataset = datasets.QM9(amount=100)
dl = DisjointLoader(dataset, batch_size=batch_size)
dl.__next__()
bl = BatchLoader(dataset, batch_size=batch_size)
bl.__next__()
def test_tud():
# Edge labels + edge attributes
dataset = datasets.TUDataset('BZR_MD', clean=False)
dl = DisjointLoader(dataset, batch_size=batch_size)
dl.__next__()
bl = BatchLoader(dataset, batch_size=batch_size)
bl.__next__()
# Node labels + node attributes + clean version
dataset = datasets.TUDataset('ENZYMES', clean=True)
dl = DisjointLoader(dataset, batch_size=batch_size)
dl.__next__()
bl = BatchLoader(dataset, batch_size=batch_size)
bl.__next__()