docstr: Tests that the input batch dimension preserved.
docstr_tok: ["Tests","that","the","input","batch","dimension","preserved","."]
code: def test_batch_dimension_preserved(self, batch_shape):
      """Tests that the input batch dimension preserved."""
      (
          vertices,
          triangles,
          attributes_dictionary,
          _,
          _,
          view_projection_matrix,
          image_height,
          image_width,
      ) = rasterization_test_utils.create_rasterizer_inputs(batch_shape)
  
      rgba = splat.rasterize_then_splat(
          vertices,
          triangles,
          attributes_dictionary,
          view_projection_matrix,
          (image_height, image_width),
          lambda x: x["attribute1"],
          backend=self._backend,
      )
  
      tensor_batch_shape = rgba.shape.as_list()[: len(batch_shape)]
      self.assertEqual(
          list(batch_shape),
          tensor_batch_shape,
          msg="Output has batch shape {0}, but expected is {1}".format(
              tensor_batch_shape, batch_shape
          ),
      )
  
file_path: graphics/tensorflow_graphics/rendering/tests/splat_test.py
imports: from absl.testing import parameterized
  import tensorflow
  from tensorflow_graphics.rendering import rasterization_backend
  from tensorflow_graphics.rendering import splat
  from tensorflow_graphics.rendering import triangle_rasterizer
  from tensorflow_graphics.rendering.tests import rasterization_test_utils
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 2 for calling the SplatTest obj's underlying member method code with example usage: obj.test_batch_dimension_preserved(batch_shape) without return types