From db1d8f916dcc6463ed8422922e624d65745aa67a Mon Sep 17 00:00:00 2001 From: Oscar Esteban Date: Thu, 4 Jul 2024 16:32:01 +0200 Subject: [PATCH] fix: meet new signature --- test/test_dipy.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/test_dipy.py b/test/test_dipy.py index 5ebebbf8..092b8c5e 100644 --- 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