code: def process_param_init(p_init, shape, rng):
      if p_init is True:
          p_init = np.asarray(rng.randn(*shape))
      return p_init
  
imports: import pytest
  import numpy
  import nnabla
  import nnabla.functions
  import nnabla.parametric_functions
  import nnabla.initializer
  from nnabla.testing import assert_allclose
  from nbla_test_utils import list_context
label_desc: Use image node_id 2 for calling a global function with example usage: process_param_init(p_init, shape, rng) and returns: p_init