Fixed tests - should work?

Author: larryshamalama

pymc/distributions/continuous.py

@@ -3689,7 +3689,13 @@ def dist(cls, x_points, pdf_points, *args, **kwargs):
     def get_moment(rv, size, x_points, pdf_points, cdf_points):
         # cdf_points argument is unused
         # moment = at.as_tensor(0.)
+        print(x_points)
+        print("")
+        for pdf in pdf_points.eval():
+            print(pdf)
+        print("")
         moment = at.sum(at.mul(x_points, pdf_points))
+        print(moment.eval())
         if not rv_size_is_none(size):
             moment = at.full(size, moment)
 

pymc/initial_point.py

@@ -272,6 +272,7 @@ def make_initial_point_expression(
         if isinstance(strategy, str):
             if strategy == "moment":
                 value = get_moment(variable)
+                print(value.eval())
             elif strategy == "prior":
                 value = variable
             else:

pymc/tests/test_distributions_moments.py

@@ -805,14 +805,14 @@ def test_categorical_moment(p, size, expected):
     "x_points, pdf_points, size, expected",
     [
         (np.array([-1, 1]), np.array([0.4, 0.6]), None, 0.2),
-        (np.array([-4, -1, 3, 9, 19]), np.array([0.1 , 0.15, 0.2 , 0.25, 0.3 ]), None, 8),
-        # (np.array([-22, -4, 0, 8, 13]), np.tile(1 / 5, 5), (5, 3), -np.ones((5, 3))),
-        # (
-        #     np.arange(-100, 10),
-        #     np.arange(1, 111) / 6105,
-        #     (2, 5, 3),
-        #     np.broadcast_to(-82 / 3, (2, 5, 3)),
-        # ),
+        (np.array([-4, -1, 3, 9, 19]), np.array([0.1 , 0.15, 0.2 , 0.25, 0.3]), None, 1.5458937198067635),
+        (np.array([-22, -4, 0, 8, 13]), np.tile(1 / 5, 5), (5, 3), np.full((5, 3), -0.14285714285714296)),
+        (
+            np.arange(-100, 10),
+            np.arange(1, 111) / 6105,
+            (2, 5, 3),
+            np.full((2, 5, 3), -27.584097859327223),
+        ),
     ],
 )
 def test_interpolated_moment(x_points, pdf_points, size, expected):