code: def assert_dataset(self, ds, expected_ds, expected_dtypes):
      actual_ds = list(tfds.as_numpy(ds))
      self.assertLen(actual_ds, len(expected_ds))
      for actual, expected in zip(actual_ds, expected_ds):
          self.assertCountEqual(
              list(actual.keys()), list(expected.keys())
          )
          for k, v in actual.items():
              np.testing.assert_array_equal(v, expected[k])
              if k in expected_dtypes:
                  self.assertEqual(v.dtype.type, expected_dtypes[k])
  
file_path: mesh/mesh_tensorflow/transformer/dataset_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from mesh_tensorflow.transformer import dataset
  import numpy
  import tensorflow.compat.v1
  import tensorflow_datasets
label_desc: Use image node_id 1 for calling the DatasetTest obj's underlying member method code with example usage: obj.assert_dataset(ds, expected_ds, expected_dtypes) without return types