code: def test_get_candidate_values():
      x = tf.constant(
          [[0, 1, 2], [3, 4, 5], [6, 7, 8]], dtype=tf.float32
      )
      candidate_ids = tf.constant([[0, 1], [0, 0], [2, 0]])
      expected_result = [
          [[0, 1, 2], [3, 4, 5]],
          [[0, 1, 2], [0, 1, 2]],
          [[6, 7, 8], [0, 1, 2]],
      ]
      actual_result = layers_utils.get_candidate_values(
          x, candidate_ids
      )
      assert np.all(expected_result == actual_result)
  
file_path: rasa/tests/utils/tensorflow/test_layers_utils.py
imports: import pytest
  import tensorflow
  import numpy
  from typing import List, Optional, Union
  import rasa.utils.tensorflow.layers_utils
label_desc: Use image node_id 4 for calling a global function with example usage: test_get_candidate_values() without return types