code: def check_ds_shape(self, ds, sequence_length):
      for k, v in tf.data.get_output_shapes(ds).items():
          feat = k.split("_")[0]
          if len(v) == 0:  # pylint:disable=g-explicit-length-test
              expected_shape = []
          elif feat in sequence_length:
              expected_shape = [sequence_length[feat]]
          else:
              expected_shape = [None]
          self.assertEqual(expected_shape, v.as_list())
  
file_path: text-to-text-transfer-transformer/t5/models/mesh_transformer_test.py
imports: from absl.testing import absltest
  import seqio
  from t5.models import mesh_transformer
  import tensorflow.compat.v1
  import tensorflow_datasets
label_desc: Use image node_id 1 for calling the MeshDatasetFnsTest obj's underlying member method code with example usage: obj.check_ds_shape(ds, sequence_length) without return types