code: def test_shape_inference(box, expected_shape):
      """Test that the shape inference is as expected."""
      assert box.shape == expected_shape
      assert box.sample().shape == expected_shape
  
file_path: gym/tests/spaces/test_box.py
imports: import re
  import warnings
  import numpy
  import pytest
  import gym.error
  from gym.spaces import Box
  from gym.spaces.box import get_inf
label_desc: Use image node_id 1 for calling a global function with example usage: test_shape_inference(box, expected_shape) without return types