code: def test_kg_remap_id(self):
      config_dict = {
          "model": "KGAT",
          "dataset": "kg_remap_id",
          "data_path": current_path,
          "load_col": None,
      }
      dataset = new_dataset(config_dict=config_dict)
      item_list = dataset.token2id("item_id", ["ib", "ic", "id"])
      entity_list = dataset.token2id(
          "entity_id", ["eb", "ec", "ed", "ee", "ea"]
      )
      assert (item_list == [1, 2, 3]).all()
      assert (entity_list == [1, 2, 3, 4, 5]).all()
      assert (dataset.inter_feat["user_id"] == [1, 2, 3]).all()
      assert (dataset.inter_feat["item_id"] == [1, 2, 3]).all()
      assert (dataset.kg_feat["head_id"] == [1, 2, 3, 4]).all()
      assert (dataset.kg_feat["tail_id"] == [5, 1, 2, 3]).all()
  
imports: import logging
  import os
  import pytest
  from recbole.config import Config
  from recbole.data import create_dataset
  from recbole.utils import init_seed
label_desc: Use image node_id 1 for calling the TestKGDataset obj's underlying member method code with example usage: obj.test_kg_remap_id() without return types