fix test for readability and variation

Author: farhanreynaldo

pymc/tests/test_distributions_moments.py

@@ -618,13 +618,13 @@ def test_discrete_uniform_moment(lower, upper, size, expected):
 @pytest.mark.parametrize(
     "p, size, expected",
     [
-        (np.arange(0.1, 0.4, 0.1), None, 3),
-        (np.arange(0.1, 0.4, 0.1), 5, np.full(5, 3)),
-        (np.full((2, 4), np.arange(0.1, 0.4, 0.1)), None, [3, 3]),
+        (np.array([0.1, 0.3, 0.6]), None, 2),
+        (np.array([0.6, 0.1, 0.3]), 5, np.full(5, 0)),
+        (np.full((2, 3), np.array([0.6, 0.1, 0.3])), None, [0, 0]),
         (
-            np.full((2, 4), np.arange(0.1, 0.4, 0.1)),
+            np.full((2, 3), np.array([0.1, 0.3, 0.6])),
             (3, 2),
-            np.full((3, 2), [3, 3]),
+            np.full((3, 2), [2, 2]),
         ),
     ],
 )