add short descriptions to tests.

dmlc · Jun 29, 2023 · 009be59 · 009be59
1 parent c3deb84
commit 009be59
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/python/common/sampling/test_sampling.py b/tests/python/common/sampling/test_sampling.py
@@ -684,6 +684,7 @@ def _test3(p, replace):
 def _test_sample_labors(hypersparse, prob):
     g, hg = _gen_neighbor_sampling_test_graph(hypersparse, False)
 
+    # test with seed nodes [0, 1]
     def _test1(p):
         subg = dgl.sampling.sample_labors(g, [0, 1], -1, prob=p)[0]
         assert subg.num_nodes() == g.num_nodes()
@@ -719,7 +720,8 @@ def _test1(p):
 
     _test1(prob)
 
-    def _test2(p):  # fanout > #neighbors
+    # test with seed nodes [0, 2]
+    def _test2(p):
         subg = dgl.sampling.sample_labors(g, [0, 2], -1, prob=p)[0]
         assert subg.num_nodes() == g.num_nodes()
         u, v = subg.edges()
@@ -753,6 +755,7 @@ def _test2(p):  # fanout > #neighbors
 
     _test2(prob)
 
+    # test with heterogenous seed nodes
     def _test3(p):
         subg = dgl.sampling.sample_labors(
             hg, {"user": [0, 1], "game": 0}, -1, prob=p