docstr: Test two consecutive calls to `step`.
docstr_tok: ["Test","two","consecutive","calls","to","`","step","`","."]
code: def testTwoConsecutiveSteps(self, batch_size, observation_shape):
      """Test two consecutive calls to `step`."""
      env = ZerosEnvironment(
          batch_size=batch_size, observation_shape=observation_shape
      )
      self.evaluate(tf.compat.v1.global_variables_initializer())
      self.evaluate(env.reset())
      self.evaluate(env.step(tf.zeros(batch_size)))
      self.evaluate(env.step(tf.zeros(batch_size)))
  
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.bandits.environments import bandit_tf_environment
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import time_step
  from tf_agents.utils import common
label_desc: Use image node_id 2 for calling the BanditTFEnvironmentTest obj's underlying member method code with example usage: obj.testTwoConsecutiveSteps(batch_size, observation_shape) without return types