code: def test_get_random_index():
      torch.manual_seed(0)
      x = torch.Tensor([1, 2, 3])
      expected = np.array([2, 0, 1])
      actual = _get_random_index(x).numpy()
      np.testing.assert_array_equal(actual, expected)
  
imports: import numpy
  import torch
  from pytorch_tabular.ssl_models.common.augmentations import _get_random_index, cutmix, mixup
label_desc: Use image node_id 1 for calling a global function with example usage: test_get_random_index() without return types