fix: meet new signature

nipreps · Jul 4, 2024 · db1d8f9 · db1d8f9
1 parent 832ca76
commit db1d8f9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/test_dipy.py b/test/test_dipy.py
@@ -23,6 +23,8 @@
 import numpy as np
 import pytest
 
+from dipy.core.gradients import gradient_table
+
 from eddymotion.model.dipy import (
     compute_exponential_covariance,
     compute_pairwise_angles,
@@ -124,7 +126,9 @@
     ],
 )
 def test_compute_pairwise_angles(bvecs1, bvecs2, closest_polarity, expected):
-    obtained = compute_pairwise_angles(bvecs1, bvecs2, closest_polarity)
+    gtab1 = gradient_table([1000] * len(bvecs1), bvecs1)
+    gtab2 = gradient_table([1000] * len(bvecs2), bvecs2)
+    obtained = compute_pairwise_angles(gtab1, gtab2, closest_polarity)
 
     assert (bvecs1.shape[-1], bvecs2.shape[-1]) == obtained.shape
     assert obtained.shape == expected.shape