Correcting Misleading Test Skip Reasons.

tests/python/pytorch/graphbolt/impl/test_gpu_cached_feature.py

@@ -15,12 +15,22 @@
     path = os.path.join(test_dir, name + ".npy")
     np.save(path, t.cpu().numpy())
     return path
 
+def _skip_condition_CachedFeature():
+    return (F._default_context_str != "gpu") or \
+        (torch.cuda.get_device_capability()[0] < 7)
+
+
+def _reson_to_skip_CachedFeature():
+    if F._default_context_str != "gpu":
+        return "GPUCachedFeature tests are available only on GPU."
+
+    return "GPUCachedFeature requires a Volta or later generation NVIDIA GPU."
+
 
 @unittest.skipIf(
-    F._default_context_str != "gpu"
-    or torch.cuda.get_device_capability()[0] < 7,
-    reason="GPUCachedFeature requires a Volta or later generation NVIDIA GPU.",
+    _skip_condition_CachedFeature(),
+    reason=_reson_to_skip_CachedFeature(),
 )
 @pytest.mark.parametrize(
     "dtype",
@@ -116,9 +126,8 @@
 
 
 @unittest.skipIf(
-    F._default_context_str != "gpu"
-    or torch.cuda.get_device_capability()[0] < 7,
-    reason="GPUCachedFeature requires a Volta or later generation NVIDIA GPU.",
+    _skip_condition_CachedFeature(),
+    reason=_reson_to_skip_CachedFeature(),
 )
 @pytest.mark.parametrize(
     "dtype",
@@ -155,9 +164,8 @@
 
 
 @unittest.skipIf(
-    F._default_context_str != "gpu"
-    or torch.cuda.get_device_capability()[0] < 7,
-    reason="GPUCachedFeature requires a Volta or later generation NVIDIA GPU.",
+    _skip_condition_CachedFeature(),
+    reason=_reson_to_skip_CachedFeature(),
 )
 @unittest.skipIf(
     not torch.ops.graphbolt.detect_io_uring(),

tests/python/pytorch/graphbolt/impl/test_hetero_cached_feature.py

@@ -1,3 +1,4 @@
+import unittest
 import backend as F
 
 import pytest
@@ -6,17 +7,17 @@
 from dgl import graphbolt as gb
 
 
+@unittest.skipIf(
+    F._default_context_str != "gpu"
+    or torch.cuda.get_device_capability()[0] < 7,
+    reason="GPUCachedFeature tests are available only on GPU."
+    if F._default_context_str != "gpu"
+    else "GPUCachedFeature requires a Volta or later generation NVIDIA GPU.",
+)
 @pytest.mark.parametrize(
     "cached_feature_type", [gb.cpu_cached_feature, gb.gpu_cached_feature]
 )
 def test_hetero_cached_feature(cached_feature_type):
-    if cached_feature_type == gb.gpu_cached_feature and (
-        F._default_context_str != "gpu"
-        or torch.cuda.get_device_capability()[0] < 7
-    ):
-        pytest.skip(
-            "GPUCachedFeature requires a Volta or later generation NVIDIA GPU."
-        )
     device = F.ctx() if cached_feature_type == gb.gpu_cached_feature else None
     pin_memory = cached_feature_type == gb.gpu_cached_feature