docstr: Tests that the observations fall into the unit circle.
docstr_tok: ["Tests","that","the","observations","fall","into","the","unit","circle","."]
code: def test_observation_validity(self, batch_size):
      """Tests that the observations fall into the unit circle."""
      env = wheel_py_environment.WheelPyEnvironment(
          delta=0.5,
          mu_base=[1.2, 1.0, 1.0, 1.0, 1.0],
          std_base=0.01 * np.ones(5),
          mu_high=50.0,
          std_high=0.01,
          batch_size=batch_size,
      )
  
      for _ in range(5):
          observation = env.reset().observation
          self.assertEqual(
              list(observation.shape),
              [batch_size] + list(env.observation_spec().shape),
          )
          for i in range(batch_size):
              self.assertLessEqual(np.linalg.norm(observation[i, :]), 1)
  
file_path: agents/tf_agents/bandits/environments/wheel_py_environment_test.py
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 wheel_py_environment
label_desc: Use image node_id 5 for calling the WheelBanditPyEnvironmentTest obj's underlying member method code with example usage: obj.test_observation_validity(batch_size) without return types