code: def test_episode_metadata(self):
      mask = {"flag0": ["episode0"], "flag1": ["episode1", "episode2"]}
      metadata = dataset_utils.episode_metadata(mask, "episode1")
      expected_metadata = {"flag0": False, "flag1": True}
      self.assertDictEqual(metadata, expected_metadata)
  
file_path: datasets/tensorflow_datasets/robomimic/dataset_utils_test.py
imports: import inspect
  import os
  from absl import flags
  from absl.testing import absltest
  from etils import epath
  import h5py
  import numpy
  from tensorflow_datasets.robomimic import dataset_utils
  import tree
label_desc: Use image node_id 4 for calling the DatasetUtilsTest obj's underlying member method code with example usage: obj.test_episode_metadata() without return types