Skip to content

Commit bb3d129

Browse files
authored
Merge pull request #211 from jhlegarreta/sty/improve-dmri-model-inst-name
STY: Honor instantiated model name in test
2 parents 1b6b681 + e344382 commit bb3d129

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

test/test_model.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -100,26 +100,26 @@ def test_average_model():
100100
data *= gtab[:, -1]
101101
dataset = DWI(dataobj=data, gradients=gtab, brainmask=mask)
102102

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)
106106

107107
# Verify that average cannot be calculated in shells with one single value
108108
with pytest.raises(RuntimeError):
109-
tmodel_mean.fit_predict(2)
109+
avgmodel_mean.fit_predict(2)
110110

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)
113113

114114
grads = list(gtab[:, -1])
115115
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))
117117

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)
120120

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)
123123

124124

125125
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)