Update tests to skip when torch is not available

yu9824 · Dec 3, 2023 · 916a72e · 916a72e
1 parent d840335
commit 916a72e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_gpu_calc.py b/tests/test_gpu_calc.py
@@ -47,6 +47,7 @@ def test_gpu_euclidian(prepare_data):
     assert np.allclose(distance_X_mean, distance_X_mean_gpu)
 
 
+@pytest.mark.skipif(device is None, reason="GPU is not available.")
 def test_gpu_manhattan(prepare_data):
     # 通常の距離行列
     X, _ = prepare_data
@@ -85,6 +86,7 @@ def test_gpu_manhattan(prepare_data):
     assert np.allclose(distance_X_mean, distance_X_mean_gpu.cpu().numpy())
 
 
+@pytest.mark.skipif(device is None, reason="GPU is not available.")
 def test_gpu_chebyshev(prepare_data):
     # 通常の距離行列
     X, _ = prepare_data