code: def test_attention_wrapper_with_gru_cell():
      mechanism = wrapper.LuongAttention(units=3)
      cell = tf.keras.layers.GRUCell(3)
      cell = wrapper.AttentionWrapper(cell, mechanism)
      memory = tf.ones([2, 5, 3])
      inputs = tf.ones([2, 3])
      mechanism.setup_memory(memory)
      initial_state = cell.get_initial_state(inputs=inputs)
      _, state = cell(inputs, initial_state)
      tf.nest.assert_same_structure(initial_state, state)
  
imports: import collections
  from packaging.version import Version
  import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.seq2seq import attention_wrapper
  from tensorflow_addons.seq2seq import basic_decoder
  from tensorflow_addons.seq2seq import sampler
label_desc: Use image node_id 27 for calling a global function with example usage: test_attention_wrapper_with_gru_cell() without return types