fix: failing test after sloppy merge

oesteban · oesteban · commit 7d4448498840 · 2025-01-23T15:41:01.000+01:00
diff --git a/test/test_model.py b/test/test_model.py
@@ -78,10 +78,6 @@ def test_trivial_model(use_mask):
 def test_average_model():
     """Check the implementation of the average DW model."""
 
-    size = (100, 100, 100, 6)
-    data = np.ones(size, dtype=float)
-    mask = np.ones(size[:3], dtype=bool)
-
     gtab = np.array(
         [
             [0, 0, 0, 0],
@@ -97,6 +93,10 @@ def test_average_model():
         ]
     )
 
+    size = (100, 100, 100, gtab.shape[0])
+    data = np.ones(size, dtype=float)
+    mask = np.ones(size[:3], dtype=bool)
+
     data *= gtab[:, -1]
     dataset = DWI(dataobj=data, gradients=gtab, brainmask=mask)
 

Original file line number	Diff line number	Diff line change
`@@ -78,10 +78,6 @@ def test_trivial_model(use_mask):`
`78`	`78`	`def test_average_model():`
`79`	`79`	`"""Check the implementation of the average DW model."""`
`80`	`80`
`81`		`- size = (100, 100, 100, 6)`
`82`		`- data = np.ones(size, dtype=float)`
`83`		`- mask = np.ones(size[:3], dtype=bool)`
`84`		`-`
`85`	`81`	`gtab = np.array(`
`86`	`82`	`[`
`87`	`83`	`[0, 0, 0, 0],`
`@@ -97,6 +93,10 @@ def test_average_model():`
`97`	`93`	`]`
`98`	`94`	`)`
`99`	`95`
	`96`	`+ size = (100, 100, 100, gtab.shape[0])`
	`97`	`+ data = np.ones(size, dtype=float)`
	`98`	`+ mask = np.ones(size[:3], dtype=bool)`
	`99`	`+`
`100`	`100`	`data *= gtab[:, -1]`
`101`	`101`	`dataset = DWI(dataobj=data, gradients=gtab, brainmask=mask)`
`102`	`102`