code: def testDynamicUnrollResetsStateOnReset(self):
      if hasattr(tf, "contrib"):
  
          class AddInputAndStateRNNCell(tf.contrib.rnn.LayerRNNCell):
              @property
              def state_size(self):
                  return tf.TensorShape([1])
  
              @property
              def output_size(self):
                  return tf.TensorShape([1])
  
              def call(self, input_, state):
                  s = input_ + state
                  return s, s
  
          self._testDynamicUnrollResetsStateOnReset(
              AddInputAndStateRNNCell
          )
  
      self._testDynamicUnrollResetsStateOnReset(
          AddInputAndStateKerasRNNCell
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tf_agents.keras_layers import dynamic_unroll_layer
label_desc: Use image node_id 5 for calling the DynamicUnrollTest obj's underlying member method code with example usage: obj.testDynamicUnrollResetsStateOnReset() and returns: tf, tf, s,  s