code: def test_xavier_init():
      set_seed(2022)
      module = nn.Linear(10, 20)
      xavier_init(module)
      assert torch.var(module.weight).item() == pytest.approx(
          2.0 / (module.in_features + module.out_features), rel=1e-1
      )
  
imports: import pytest
  import torch
  import torch.nn
  from kale.utils.initialize_nn import bias_init, xavier_init
  from kale.utils.seed import set_seed
label_desc: Use image node_id 1 for calling a global function with example usage: test_xavier_init() without return types