Add a test

flatironinstitute · Oct 11, 2023 · 75f7c6a · 75f7c6a
1 parent 5a7904e
commit 75f7c6a
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_errors.py b/tests/test_errors.py
@@ -157,3 +157,14 @@ def test_t1_negative_output_dims() -> None:
         ValueError, match="Got output_shape that was not positive integer"
     ):
         pytorch_finufft.functional.finufft_type1.apply(points, values, (10, -2))
+
+# dependencies
+def test_cuda_finufft_not_installed():
+    pytest.importorskip("finufft")
+
+    points = torch.rand(10, dtype=torch.float64).to('cpu')
+    values = torch.randn(10, dtype=torch.complex128).to('cpu')
+
+    with pytest.raises(RuntimeError, match="cufinufft failed to import"):
+        pytorch_finufft.functional.finufft_type1.apply(points, values, 10)
+