def METHOD_NAME(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]