class_docstr: load the FB15K
docstr: Test data reader.
  :param batch_size: Todo: batch_size > 1.
  :return: None
code: def test_data_batch(self, batch_size=None):
      """
      Test data reader.
      :param batch_size: Todo: batch_size > 1.
      :return: None
      """
      for i in range(self.test_num):
          data = np.array(self._triple_test[i])
          data = data.reshape((-1))
          yield [data]
  
file_path: PGL/legacy/examples/pgl-ke/data_loader.py
imports: import os
  import numpy
  from collections import defaultdict
  from pgl.utils.logger import log
label_desc: Use image node_id 2 for calling the KGLoader obj's underlying member method code with example usage: obj.test_data_batch(batch_size) without return types