code: def test_batch_remap_label_ids(
      self, before, after, expected_old_id_to_new_id
  ):
      unordered_labels = np.array(before)
      expected = np.array(after)
      (
          batch_remapped,
          batch_old_id_to_new_id,
      ) = data_utils.batch_remap_label_ids(unordered_labels)
      np.testing.assert_array_equal(batch_remapped, expected)
      self.assertListEqual(
          batch_old_id_to_new_id, expected_old_id_to_new_id
      )
  
file_path: neural-structured-learning/research/neural_clustering/utils/data_utils_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from neural_clustering.utils import data_utils
  import numpy
label_desc: Use image node_id 2 for calling the UtilsTest obj's underlying member method code with example usage: obj.test_batch_remap_label_ids(before, after, expected_old_id_to_new_id) without return types