@@ -100,26 +100,26 @@ def test_average_model():
100
100
data *= gtab [:, - 1 ]
101
101
dataset = DWI (dataobj = data , gradients = gtab , brainmask = mask )
102
102
103
- tmodel_mean = model .AverageDWIModel (dataset , stat = "mean" )
104
- tmodel_mean_full = model .AverageDWIModel (dataset , stat = "mean" , atol_low = 2000 , atol_high = 2000 )
105
- tmodel_median = model .AverageDWIModel (dataset )
103
+ avgmodel_mean = model .AverageDWIModel (dataset , stat = "mean" )
104
+ avgmodel_mean_full = model .AverageDWIModel (dataset , stat = "mean" , atol_low = 2000 , atol_high = 2000 )
105
+ avgmodel_median = model .AverageDWIModel (dataset )
106
106
107
107
# Verify that average cannot be calculated in shells with one single value
108
108
with pytest .raises (RuntimeError ):
109
- tmodel_mean .fit_predict (2 )
109
+ avgmodel_mean .fit_predict (2 )
110
110
111
- assert np .allclose (tmodel_mean .fit_predict (3 ), 1000 )
112
- assert np .allclose (tmodel_median .fit_predict (3 ), 1000 )
111
+ assert np .allclose (avgmodel_mean .fit_predict (3 ), 1000 )
112
+ assert np .allclose (avgmodel_median .fit_predict (3 ), 1000 )
113
113
114
114
grads = list (gtab [:, - 1 ])
115
115
del grads [1 ]
116
- assert np .allclose (tmodel_mean_full .fit_predict (1 ), np .mean (grads ))
116
+ assert np .allclose (avgmodel_mean_full .fit_predict (1 ), np .mean (grads ))
117
117
118
- tmodel_mean_2000 = model .AverageDWIModel (dataset , stat = "mean" , atol_low = 1100 )
119
- tmodel_median_2000 = model .AverageDWIModel (dataset , atol_low = 1100 )
118
+ avgmodel_mean_2000 = model .AverageDWIModel (dataset , stat = "mean" , atol_low = 1100 )
119
+ avgmodel_median_2000 = model .AverageDWIModel (dataset , atol_low = 1100 )
120
120
121
- assert np .allclose (tmodel_mean_2000 .fit_predict (9 ), gtab [3 :- 1 , - 1 ].mean ())
122
- assert np .allclose (tmodel_median_2000 .fit_predict (9 ), 1000 )
121
+ assert np .allclose (avgmodel_mean_2000 .fit_predict (9 ), gtab [3 :- 1 , - 1 ].mean ())
122
+ assert np .allclose (avgmodel_median_2000 .fit_predict (9 ), 1000 )
123
123
124
124
125
125
@pytest .mark .parametrize (
0 commit comments