code: def test_broadcast_common_batch_shape(self):
      x = tf.zeros([3, 4])
      y = tf.zeros([2, 1, 3, 10])
      z = tf.zeros([])
      x, y, z = tff.utils.broadcast_common_batch_shape(x, y, z)
      with self.subTest("ShapeX"):
          self.assertAllEqual(x, np.zeros([2, 1, 3, 4]))
      with self.subTest("ShapeY"):
          self.assertAllEqual(y, np.zeros([2, 1, 3, 10]))
      with self.subTest("ShapeZ"):
          self.assertAllEqual(z, np.zeros([2, 1, 3]))
  
file_path: tf-quant-finance/tf_quant_finance/utils/shape_utils_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  import tf_quant_finance
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 7 for calling the ShapeUtilsTest obj's underlying member method code with example usage: obj.test_broadcast_common_batch_shape() without return types