code: def test_layer_output(attention_cls):
      dummy_data = DummyData()
      attention = attention_cls(dummy_data.units, dummy_data.memory)
      score = attention([dummy_data.query, dummy_data.state])
  
      assert len(score) == 2
      assert score[0].shape == (dummy_data.batch, dummy_data.timestep)
      assert score[1].shape == (dummy_data.batch, dummy_data.timestep)
  
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 3 for calling a global function with example usage: test_layer_output(attention_cls) without return types