[Misc] Move dataset_tests to an independent directory. (#6617)

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>

tests/integration/test_data.py

@@ -211,3 +211,65 @@ def test_as_graphpred():
     assert len(new_ds) == 1000
     assert new_ds.num_tasks == 1
     assert new_ds.num_classes == 2
+
+
+@unittest.skipIf(
+    dgl.backend.backend_name != "pytorch", reason="ogb only supports pytorch"
+)
+def test_as_linkpred_ogb():
+    from ogb.linkproppred import DglLinkPropPredDataset
+
+    ds = data.AsLinkPredDataset(
+        DglLinkPropPredDataset("ogbl-collab"), split_ratio=None, verbose=True
+    )
+    # original dataset has 46329 test edges
+    assert ds.test_edges[0][0].shape[0] == 46329
+    # force generate new split
+    ds = data.AsLinkPredDataset(
+        DglLinkPropPredDataset("ogbl-collab"),
+        split_ratio=[0.7, 0.2, 0.1],
+        verbose=True,
+    )
+    assert ds.test_edges[0][0].shape[0] == 235812
+
+
+@unittest.skipIf(
+    dgl.backend.backend_name != "pytorch", reason="ogb only supports pytorch"
+)
+@unittest.skipIf(dgl.backend.backend_name == "mxnet", reason="Skip MXNet")
+def test_as_nodepred_ogb():
+    from ogb.nodeproppred import DglNodePropPredDataset
+
+    ds = data.AsNodePredDataset(
+        DglNodePropPredDataset("ogbn-arxiv"), split_ratio=None, verbose=True
+    )
+    split = DglNodePropPredDataset("ogbn-arxiv").get_idx_split()
+    train_idx, val_idx, test_idx = split["train"], split["valid"], split["test"]
+    assert F.array_equal(ds.train_idx, F.tensor(train_idx))
+    assert F.array_equal(ds.val_idx, F.tensor(val_idx))
+    assert F.array_equal(ds.test_idx, F.tensor(test_idx))
+    # force generate new split
+    ds = data.AsNodePredDataset(
+        DglNodePropPredDataset("ogbn-arxiv"),
+        split_ratio=[0.7, 0.2, 0.1],
+        verbose=True,
+    )
+
+
+@unittest.skipIf(
+    dgl.backend.backend_name != "pytorch", reason="ogb only supports pytorch"
+)
+def test_as_graphpred_ogb():
+    from ogb.graphproppred import DglGraphPropPredDataset
+
+    ds = data.AsGraphPredDataset(
+        DglGraphPropPredDataset("ogbg-molhiv"), split_ratio=None, verbose=True
+    )
+    assert len(ds.train_idx) == 32901
+    # force generate new split
+    ds = data.AsGraphPredDataset(
+        DglGraphPropPredDataset("ogbg-molhiv"),
+        split_ratio=[0.6, 0.2, 0.2],
+        verbose=True,
+    )
+    assert len(ds.train_idx) == 24676

tests/python/common/data/test_data.py

@@ -1758,29 +1758,6 @@ def test_as_nodepred2():
     assert len(ds.train_idx) == int(ds[0].num_nodes("Personen") * 0.1)
 
 
-@unittest.skipIf(
-    dgl.backend.backend_name != "pytorch", reason="ogb only supports pytorch"
-)
-@unittest.skipIf(dgl.backend.backend_name == "mxnet", reason="Skip MXNet")
-def test_as_nodepred_ogb():
-    from ogb.nodeproppred import DglNodePropPredDataset
-
-    ds = data.AsNodePredDataset(
-        DglNodePropPredDataset("ogbn-arxiv"), split_ratio=None, verbose=True
-    )
-    split = DglNodePropPredDataset("ogbn-arxiv").get_idx_split()
-    train_idx, val_idx, test_idx = split["train"], split["valid"], split["test"]
-    assert F.array_equal(ds.train_idx, F.tensor(train_idx))
-    assert F.array_equal(ds.val_idx, F.tensor(val_idx))
-    assert F.array_equal(ds.test_idx, F.tensor(test_idx))
-    # force generate new split
-    ds = data.AsNodePredDataset(
-        DglNodePropPredDataset("ogbn-arxiv"),
-        split_ratio=[0.7, 0.2, 0.1],
-        verbose=True,
-    )
-
-
 @unittest.skipIf(
     F._default_context_str == "gpu",
     reason="Datasets don't need to be tested on GPU.",
@@ -1810,26 +1787,6 @@ def test_as_linkpred():
     assert 4000 < ds.test_edges[1][0].shape[0] <= 4224
 
 
-@unittest.skipIf(
-    dgl.backend.backend_name != "pytorch", reason="ogb only supports pytorch"
-)
-def test_as_linkpred_ogb():
-    from ogb.linkproppred import DglLinkPropPredDataset
-
-    ds = data.AsLinkPredDataset(
-        DglLinkPropPredDataset("ogbl-collab"), split_ratio=None, verbose=True
-    )
-    # original dataset has 46329 test edges
-    assert ds.test_edges[0][0].shape[0] == 46329
-    # force generate new split
-    ds = data.AsLinkPredDataset(
-        DglLinkPropPredDataset("ogbl-collab"),
-        split_ratio=[0.7, 0.2, 0.1],
-        verbose=True,
-    )
-    assert ds.test_edges[0][0].shape[0] == 235812
-
-
 @unittest.skipIf(
     F._default_context_str == "gpu",
     reason="Datasets don't need to be tested on GPU.",
@@ -1991,25 +1948,6 @@ def test_as_graphpred_reprocess():
     assert len(ds.train_idx) == int(len(ds) * 0.1)
 
 
-@unittest.skipIf(
-    dgl.backend.backend_name != "pytorch", reason="ogb only supports pytorch"
-)
-def test_as_graphpred_ogb():
-    from ogb.graphproppred import DglGraphPropPredDataset
-
-    ds = data.AsGraphPredDataset(
-        DglGraphPropPredDataset("ogbg-molhiv"), split_ratio=None, verbose=True
-    )
-    assert len(ds.train_idx) == 32901
-    # force generate new split
-    ds = data.AsGraphPredDataset(
-        DglGraphPropPredDataset("ogbg-molhiv"),
-        split_ratio=[0.6, 0.2, 0.2],
-        verbose=True,
-    )
-    assert len(ds.train_idx) == 24676
-
-
 if __name__ == "__main__":
     test_minigc()
     test_gin()