Merge pull request #16 from yu9824/dev

Update tests to skip when torch is not available
yu9824 · Dec 3, 2023 · 2df6010 · 2df6010
2 parents f8ed5e3 + 916a72e
commit 2df6010
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