code: def _check_equal(self, a: dm_env.TimeStep, b: dm_env.TimeStep):
      self.assertEqual(a.reward, b.reward)
      self.assertEqual(a.discount, b.discount)
      self.assertEqual(a.step_type, b.step_type)
      np.testing.assert_array_equal(a.observation, b.observation)
  
file_path: acme/acme/agents/tf/mcts/models/simulator_test.py
imports: from acme.agents.tf.mcts.models import simulator
  from bsuite.environments import catch
  import dm_env
  import numpy
  from absl.testing import absltest
label_desc: Use image node_id 1 for calling the SimulatorTest obj's underlying member method code with example usage: obj._check_equal(a, b) without return types