Fix shape/type mismatch in TestScanInplaceOptimizer.test_inplace3

brandonwillard · brandonwillard · commit e59cef1883a1 · 2022-10-21T12:41:34.000-05:00
diff --git a/tests/scan/test_rewriting.py b/tests/scan/test_rewriting.py
@@ -1101,7 +1101,7 @@ def test_inplace3(self):
         outputs, updates = scan(
             lambda x, y: (x + asarrayX(1), y + asarrayX(1)), [], [x0, x1], n_steps=3
         )
-        x0 = asarrayX(np.zeros((3,)))
+        x0 = asarrayX(np.zeros((4,)))
         x0[0] = vx0
         x0 = at.constant(x0)
 

Original file line number	Diff line number	Diff line change
`@@ -1101,7 +1101,7 @@ def test_inplace3(self):`
`1101`	`1101`	`outputs, updates = scan(`
`1102`	`1102`	`lambda x, y: (x + asarrayX(1), y + asarrayX(1)), [], [x0, x1], n_steps=3`
`1103`	`1103`	`)`
`1104`		`- x0 = asarrayX(np.zeros((3,)))`
	`1104`	`+ x0 = asarrayX(np.zeros((4,)))`
`1105`	`1105`	`x0[0] = vx0`
`1106`	`1106`	`x0 = at.constant(x0)`
`1107`	`1107`