code: def test_array_shape_static_checks():
      assert (
          beam_search_decoder._check_static_batch_beam_maybe(
              tf.TensorShape([None, None, None]), 3, 5
          )
          is True
      )
  
      assert (
          beam_search_decoder._check_static_batch_beam_maybe(
              tf.TensorShape([15, None, None]), 3, 5
          )
          is True
      )
      assert (
          beam_search_decoder._check_static_batch_beam_maybe(
              tf.TensorShape([16, None, None]), 3, 5
          )
          is False
      )
      assert (
          beam_search_decoder._check_static_batch_beam_maybe(
              tf.TensorShape([3, 5, None]), 3, 5
          )
          is True
      )
      assert (
          beam_search_decoder._check_static_batch_beam_maybe(
              tf.TensorShape([3, 6, None]), 3, 5
          )
          is False
      )
      assert (
          beam_search_decoder._check_static_batch_beam_maybe(
              tf.TensorShape([5, 3, None]), 3, 5
          )
          is False
      )
  
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 10 for calling a global function with example usage: test_array_shape_static_checks() without return types