code: def test_weight_normalization_forward_backward(
      eps, dim, shape, seed, ctx, func_name
  ):
      from nbla_test_utils import function_tester
  
      rng = np.random.RandomState(seed)
      w = rng.randn(*shape)
      g = rng.randn(shape[dim])
      inputs = [w, g]
      func_args = [dim, eps]
      function_tester(
          rng,
          F.weight_normalization,
          ref_weight_normalization,
          inputs,
          ctx=ctx,
          func_name=func_name,
          func_args=func_args,
          backward=[True, True],
          disable_half_test=False,
          atol_b=3e-3,
          atol_accum=3e-3,
      )
  
file_path: nnabla/python/test/function/test_weight_normalization.py
imports: import pytest
  import numpy
  import nnabla.functions
  from nbla_test_utils import list_context
label_desc: Use image node_id 2 for calling a global function with example usage: test_weight_normalization_forward_backward(eps, dim, shape, seed, ctx, func_name) without return types