docstr: Tests that each batch has the fixed length.
docstr_tok: ["Tests","that","each","batch","has","the","fixed","length","."]
code: def test_batched_dataset_fixed_length(self):
      """Tests that each batch has the fixed length."""
      batch_size = 2
      fixed_pad_length = 77
      for batch in self.dataset.get_batches(
          batch_size=batch_size,
          shuffle=False,
          fixed_pad_length=fixed_pad_length,
      ).take(1):
          length = batch["token_ids"].numpy().shape[-1]
          self.assertEqual(fixed_pad_length, length)
  
file_path: flax/examples/sst2/input_pipeline_test.py
imports: import os
  import pathlib
  import tempfile
  from absl.testing import absltest
  import tensorflow_datasets
  import input_pipeline
  import vocabulary
label_desc: Use image node_id 6 for calling the InputPipelineTest obj's underlying member method code with example usage: obj.test_batched_dataset_fixed_length() without return types