code: def test_gather_tree_from_array_scalar():
      _test_gather_tree_from_array()
  
imports: import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.seq2seq import attention_wrapper
  from tensorflow_addons.seq2seq import beam_search_decoder, gather_tree
label_desc: Use image node_id 3 for calling a global function with example usage: test_gather_tree_from_array_scalar() without return types