Separate transitive shape checks into separate asserts

Author: ricardoV94

pymc/tests/test_distributions_moments.py

@@ -161,7 +161,8 @@ def assert_moment_is_expected(model, expected, check_finite_logp=True):
     except NotImplementedError:
         random_draw = moment
 
-    assert moment.shape == expected.shape == random_draw.shape
+    assert moment.shape == expected.shape
+    assert expected.shape == random_draw.shape
     assert np.allclose(moment, expected)
 
     if check_finite_logp:

pymc/tests/test_distributions_random.py

@@ -375,7 +375,8 @@ def check_rv_size(self):
             pymc_rv = self.pymc_dist.dist(**self.pymc_dist_params, size=size)
             expected_symbolic = tuple(pymc_rv.shape.eval())
             actual = pymc_rv.eval().shape
-            assert actual == expected_symbolic == expected
+            assert actual == expected_symbolic
+            assert expected_symbolic == expected
 
         # test multi-parameters sampling for univariate distributions (with univariate inputs)
         if (