code: def testStateIncrementPolicy(self):
      policy = StateIncrementPolicy(
          self._time_step_spec, self._action_spec
      )
      policy_state = policy.get_initial_state(1)
      step = policy.action(self._time_step, policy_state)
      self.assertEqual(1, self.evaluate(step.action))
      step = policy.action(self._time_step, step.state)
      self.assertEqual(2, self.evaluate(step.action))
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  from six.moves import range
  import tensorflow
  from tf_agents.policies import temporal_action_smoothing
  from tf_agents.policies import tf_policy
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import policy_step
  from tf_agents.trajectories import time_step
  from tf_agents.utils import test_utils
label_desc: Use image node_id 3 for calling the TemporalActionSmoothingTest obj's underlying member method code with example usage: obj.testStateIncrementPolicy() without return types