code: def test_randomized_argmax(self):
      self._test_randomized_argmax(
          torch.tensor([1, 2, 3, 2, 3, 1, 3]), {2, 4, 6}
      )
      self._test_randomized_argmax(
          torch.tensor(
              [
                  1,
                  torch.tensor(float("inf")),
                  3,
                  2,
                  3,
                  torch.tensor(float("inf")),
                  3,
              ]
          ),
          {1, 5},
      )
      self._test_randomized_argmax(
          torch.tensor(
              [
                  torch.tensor(float("inf")),
                  torch.tensor(float("inf")),
                  torch.tensor(float("inf")),
              ]
          ),
          {0, 1, 2},
      )
      self._test_randomized_argmax(
          torch.tensor([1, 2, 3, 2, 3, 1, 5]), {6}
      )
  
imports: import unittest
  from io import BytesIO
  from itertools import cycle
  from unittest import mock
  import numpy
  import numpy.testing
  import torch
  from parameterized.parameterized import parameterized
  from pytorch_lightning import seed_everything
  from reagent.mab.mab_algorithm import get_arm_indices, place_values_at_indices, randomized_argmax, reindex_multiple_tensors
  from reagent.mab.simulation import BernoilliMAB, compare_bandit_algos, multiple_evaluations_bandit_algo, single_evaluation_bandit_algo
  from reagent.mab.thompson_sampling import BaseThompsonSampling, BernoulliBetaThompson, NormalGammaThompson
  from reagent.mab.ucb import BaseUCB, MetricUCB, UCB1, UCBTuned
label_desc: Use image node_id 7 for calling the TestMAButils obj's underlying member method code with example usage: obj.test_randomized_argmax() without return types