code: def test_cases():
      return parameterized.named_parameters(
          dict(
              testcase_name="_observation_[5]_action_[3]_batch_1",
              observation_shape=[5],
              action_shape=[3],
              batch_size=1,
              seed=12345,
          ),
          dict(
              testcase_name="_observation_[3]_action_[5]_batch_2",
              observation_shape=[3],
              action_shape=[5],
              batch_size=2,
              seed=98765,
          ),
      )
  
file_path: agents/tf_agents/bandits/environments/drifting_linear_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
  import tensorflow_probability
  from tf_agents.bandits.environments import drifting_linear_environment
label_desc: Use image node_id 1 for calling a global function with example usage: test_cases() and returns: parameterized