code: def test_kaiming(self, nn_init):
      callback = init.KaimingNormal(
          a=1, mode="test", nonlinearity="test2"
      )
      mock = MagicMock()
      callback.initialiser(mock)
      nn_init.kaiming_normal_.assert_called_once_with(
          mock.weight.data, a=1, mode="test", nonlinearity="test2"
      )
  
      callback = init.KaimingUniform(
          a=1, mode="test", nonlinearity="test2"
      )
      mock = MagicMock()
      callback.initialiser(mock)
      nn_init.kaiming_uniform_.assert_called_once_with(
          mock.weight.data, a=1, mode="test", nonlinearity="test2"
      )
  
file_path: torchbearer/tests/callbacks/test_init.py
imports: import torch
  from unittest import TestCase
  from mock import MagicMock, patch
  import torchbearer
  import torchbearer.callbacks.init
label_desc: Use image node_id 1 for calling the TestSimpleInits obj's underlying member method code with example usage: obj.test_kaiming(nn_init) without return types