code: def test_keras_rnn():
      """Tests that NASCell works with keras RNN layer."""
      cell = NASCell(10)
      seq_input = tf.convert_to_tensor(
          np.random.rand(2, 3, 5), name="seq_input", dtype=tf.float32
      )
      rnn_layer = keras.layers.RNN(cell=cell)
      rnn_outputs = rnn_layer(seq_input)
      assert rnn_outputs.shape == (2, 10)
  
file_path: addons/tensorflow_addons/rnn/tests/nas_cell_test.py
imports: import numpy
  import tensorflow
  import tensorflow.keras
  from tensorflow_addons.rnn import NASCell
label_desc: Use image node_id 3 for calling a global function with example usage: test_keras_rnn() without return types