code: def test_factored_joint_mvn_diag_full(self):
      batch_shape = [3, 2]
  
      mvn1 = mvn_diag.MultivariateNormalDiag(
          loc=tf.zeros(batch_shape + [3]),
          scale_diag=tf.ones(batch_shape + [3]),
      )
  
      mvn2 = mvn_tril.MultivariateNormalTriL(
          loc=tf.ones(batch_shape + [2]),
          scale_tril=(
              tf.ones(batch_shape + [2, 2])
              * tf.linalg.cholesky([[5.0, -2], [-2, 3.1]])
          ),
      )
  
      joint = sts_util.factored_joint_mvn([mvn1, mvn2])
      self.assertEqual(
          self.evaluate(joint.event_shape_tensor()),
          self.evaluate(
              mvn1.event_shape_tensor() + mvn2.event_shape_tensor()
          ),
      )
  
      joint_mean_ = self.evaluate(joint.mean())
      self.assertAllEqual(
          joint_mean_[..., :3], self.evaluate(mvn1.mean())
      )
      self.assertAllEqual(
          joint_mean_[..., 3:], self.evaluate(mvn2.mean())
      )
  
      joint_cov_ = self.evaluate(joint.covariance())
      self.assertAllEqual(
          joint_cov_[..., :3, :3], self.evaluate(mvn1.covariance())
      )
      self.assertAllEqual(
          joint_cov_[..., 3:, 3:], self.evaluate(mvn2.covariance())
      )
  
imports: from absl.testing import parameterized
  import numpy
  import pandas
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.distributions import mvn_tril
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.sts.components import local_level
  from tensorflow_probability.python.sts.internal import missing_values_util
  from tensorflow_probability.python.sts.internal import util
label_desc: Use image node_id 1 for calling the MultivariateNormalUtilsTest obj's underlying member method code with example usage: obj.test_factored_joint_mvn_diag_full() without return types