code: def test_break(self):
      import numpy as np
      from torchbearer.callbacks.init import ZeroBias
  
      np.random.seed(7)
      torch.manual_seed(7)
  
      model = torch.nn.Sequential(
          torch.nn.Conv2d(1, 1, 1),
      )
  
      with patch(
          "torchbearer.callbacks.lsuv.LSUV.apply_weights_correction"
      ) as awc:
          state = {torchbearer.MODEL: model}
          data = torch.rand(2, 1, 2, 2)
          ZeroBias(model.modules()).on_init(
              state
          )  # LSUV expects biases to be zero
          init.LsuvInit(
              data, std_tol=1e-20, max_attempts=0, do_orthonorm=False
          ).on_init(state)
  
          # torchbearer.callbacks.lsuv.apply_weights_correction = old_fun
          self.assertTrue(awc.call_count == 2)
  
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 2 for calling the TestLsuv obj's underlying member method code with example usage: obj.test_break() without return types