Refactor test_zsn_dims_shape

AlexAndorra · AlexAndorra · commit cf5b384271c6 · 2022-09-29T15:42:39.000+02:00
diff --git a/pymc/tests/distributions/test_multivariate.py b/pymc/tests/distributions/test_multivariate.py
@@ -1016,8 +1016,8 @@ def test_mv_normal_moment(self, mu, cov, size, expected):
         "shape, zerosum_axes, expected",
         [
             ((2, 5), None, np.zeros((2, 5))),
-            ((2, 5, 6), None, np.zeros((2, 5, 6))),
-            ((2, 5, 6), (0, 1), np.zeros((2, 5, 6))),
+            ((2, 5, 6), 2, np.zeros((2, 5, 6))),
+            ((2, 5, 6), 3, np.zeros((2, 5, 6))),
         ],
     )
     def test_zerosum_normal_moment(self, shape, zerosum_axes, expected):

Original file line number	Diff line number	Diff line change
`@@ -1016,8 +1016,8 @@ def test_mv_normal_moment(self, mu, cov, size, expected):`
`1016`	`1016`	`"shape, zerosum_axes, expected",`
`1017`	`1017`	`[`
`1018`	`1018`	`((2, 5), None, np.zeros((2, 5))),`
`1019`		`- ((2, 5, 6), None, np.zeros((2, 5, 6))),`
`1020`		`- ((2, 5, 6), (0, 1), np.zeros((2, 5, 6))),`
	`1019`	`+ ((2, 5, 6), 2, np.zeros((2, 5, 6))),`
	`1020`	`+ ((2, 5, 6), 3, np.zeros((2, 5, 6))),`
`1021`	`1021`	`],`
`1022`	`1022`	`)`
`1023`	`1023`	`def test_zerosum_normal_moment(self, shape, zerosum_axes, expected):`