code: def testBanditPolicyUniformMask(self):
      dims = (10, 1)
      input_tensor = tf.fill(dims, value=_GREEDY)
      # Overwrite some values with UNIFORM.
      mask_idx = range(dims[0])[1 : dims[0] : 2]
      mask = [
          [True if idx in mask_idx else False] for idx in range(dims[0])
      ]
      expected = [
          [_UNIFORM if mask_value[0] else _GREEDY]
          for mask_value in mask
      ]
      result = policy_utilities.bandit_policy_uniform_mask(
          input_tensor, mask
      )
      self.assertAllEqual(result, expected)
  
file_path: agents/tf_agents/bandits/policies/policy_utilities_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import tensorflow
  from tf_agents.policies import utils
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import policy_step
  from tf_agents.utils import test_utils
label_desc: Use image node_id 7 for calling the PolicyUtilitiesTest obj's underlying member method code with example usage: obj.testBanditPolicyUniformMask() without return types