code: def test_input(self):
      mixup = Mixup()
      X = torch.Tensor([[1.0, 2.0, 3.0], [6.0, 7.0, 8.0]])
      Y_true = torch.Tensor([0.0, 1.0])
  
      state = {torchbearer.X: X, torchbearer.Y_TRUE: Y_true}
  
      mixup.on_sample(state)
      self.assertTrue(
          (
              torch.eq(
                  state[torchbearer.X],
                  X * state[torchbearer.MIXUP_LAMBDA]
                  + X[state[torchbearer.MIXUP_PERMUTATION], :]
                  * (1 - state[torchbearer.MIXUP_LAMBDA]),
              )
          ).all()
      )
  
file_path: torchbearer/tests/callbacks/test_mixup.py
imports: from unittest import TestCase
  from mock import patch, call, MagicMock
  import torch
  import torchbearer
  from torchbearer.callbacks import Mixup, MixupAcc
label_desc: Use image node_id 1 for calling the TestMixupInputs obj's underlying member method code with example usage: obj.test_input() without return types