code: def test_cutmix():
      torch.manual_seed(0)
      np.random.seed(0)
      x = torch.Tensor([[1, 1], [2, 2], [3, 3]])
      lam = 0.5
      expected = torch.Tensor(
          [[1.0, 1.0], [2.0, 2.0], [2.0, 3.0]]
      ).numpy()
      actual = cutmix(batch={"x": x}, lam=lam)
      np.testing.assert_array_equal(actual["x"].numpy(), 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 3 for calling a global function with example usage: test_cutmix() without return types