code: def _assert_recording_two_steps(
      env: TrajectoryRecordingWrapper,
  ) -> None:
      """Perform two steps and assert that both are recorded correctly, regardless of any
      step-skipping that might be taking place inside of the env."""
  
      # Step the env twice
      maze_actions = []
      env.reset()
      for i in range(2):
          action = env.action_space.sample()
          maze_actions.append(action)
          env.step(action)
  
      # Check that the correct action contents have been recorded
      assert len(env.episode_record.step_records) == 2
      for i in range(2):
          action_taken = maze_actions[i]
          action_recorded = env.episode_record.step_records[
              i
          ].maze_action
          assert list(action_taken.keys()) == list(
              action_recorded.keys()
          )
          for key in action_taken:
              assert np.all(action_taken[key] == action_recorded[key])
  
imports: from abc import ABC
  from copy import deepcopy
  from typing import Any, Dict, List, Type
  import numpy
  from maze.core.env.base_env_events import BaseEnvEvents
  from maze.core.events.pubsub import Pubsub
  from maze.core.rendering.renderer import Renderer
  from maze.core.trajectory_recording.records.trajectory_record import StateTrajectoryRecord
  from maze.core.trajectory_recording.writers.trajectory_writer import TrajectoryWriter
  from maze.core.trajectory_recording.writers.trajectory_writer_registry import TrajectoryWriterRegistry
  from maze.core.wrappers.trajectory_recording_wrapper import TrajectoryRecordingWrapper
  from maze.test.shared_test_utils.dummy_wrappers.step_skip_in_step_wrapper import StepSkipInStepWrapper
  from maze.test.shared_test_utils.dummy_wrappers.step_skip_in_reset_wrapper import StepSkipInResetWrapper
  from maze.test.shared_test_utils.dummy_env.agents.dummy_policy import DummyGreedyPolicy
  from maze.test.shared_test_utils.dummy_env.dummy_core_env import DummyCoreEnvironment
  from maze.test.shared_test_utils.dummy_env.dummy_maze_env import DummyEnvironment
  from maze.test.shared_test_utils.dummy_env.dummy_renderer import DummyRenderer
  from maze.test.shared_test_utils.dummy_env.dummy_struct_env import DummyStructuredEnvironment
  from maze.test.shared_test_utils.dummy_env.reward.base import RewardAggregator
  from maze.test.shared_test_utils.dummy_env.space_interfaces.action_conversion.dict import DictActionConversion
  from maze.test.shared_test_utils.dummy_env.space_interfaces.observation_conversion.dict import ObservationConversion
  from maze.test.shared_test_utils.helper_functions import build_dummy_maze_env
label_desc: Use image node_id 3 for calling a global function with example usage: _assert_recording_two_steps(env) without return types