code: def test_raises_invalid_action_spec(self):
      action_spec = tensor_spec.BoundedTensorSpec(
          (1, 1), tf.int32, 0, 4
      )
      with self.assertRaisesRegexp(ValueError, "at most one dimension"):
          tf_wrappers.OneHotActionWrapper(
              _build_test_env(action_spec=action_spec)
          )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing.absltest import mock
  import tensorflow
  from tf_agents.environments import random_tf_environment
  from tf_agents.environments import tf_wrappers
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import time_step
  from tf_agents.utils import test_utils
label_desc: Use image node_id 3 for calling the OneHotActionWrapperTest obj's underlying member method code with example usage: obj.test_raises_invalid_action_spec() without return types