class_docstr: Base class for CPU/GPU rasterization backend tests.
class_docstr_tok: ["Base","class","for","CPU\/GPU","rasterization","backend","tests","."]
docstr: Tests that supported backends return correct shape.
docstr_tok: ["Tests","that","supported","backends","return","correct","shape","."]
code: def test_rasterizer_return_correct_batch_shapes(
      self, shapes, dtypes, enable_cull_face
  ):
      """Tests that supported backends return correct shape."""
      placeholders = self._create_placeholders(shapes, dtypes)
      frame_buffer = rasterization_backend.rasterize(
          placeholders[0],
          placeholders[1],
          placeholders[2],
          (self.IMAGE_WIDTH, self.IMAGE_HEIGHT),
          enable_cull_face,
          self._num_layers,
          self._backend,
      ).layer(0)
      batch_size = shapes[0][0]
      self.assertEqual(
          [batch_size],
          frame_buffer.triangle_id.get_shape().as_list()[:-3],
      )
      self.assertEqual(
          [batch_size],
          frame_buffer.foreground_mask.get_shape().as_list()[:-3],
      )
  
file_path: graphics/tensorflow_graphics/rendering/tests/rasterization_backend_test_base.py
imports: from absl.testing import parameterized
  import tensorflow
  from tensorflow_graphics.rendering import rasterization_backend
  from tensorflow_graphics.rendering.tests import rasterization_test_utils
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 4 for calling the RasterizationBackendTestBase obj's underlying member method code with example usage: obj.test_rasterizer_return_correct_batch_shapes(shapes, dtypes, enable_cull_face) without return types