code: def test_kernel_builder(self, builder_fn, expected_params):
      layer = nn.Dense(5, kernel_init=builder_fn())
      params = layer.init(random.key(42), jnp.empty((3, 2)))["params"]
      np.testing.assert_allclose(params["kernel"], expected_params)
  
file_path: flax/tests/linen/initializers_test.py
imports: import jax
  import jax.numpy
  import numpy
  from absl.testing import absltest, parameterized
  from jax import random
  from flax import linen
  from flax.linen import initializers
label_desc: Use image node_id 2 for calling the InitializersTest obj's underlying member method code with example usage: obj.test_kernel_builder(builder_fn, expected_params) without return types