code: def test_transform(self):
      def f():
          w1 = base.get_parameter("w1", [], init=jnp.zeros)
          w2 = base.get_parameter("w2", [], init=jnp.ones)
          return w1, w2
  
      f = transform.transform(f)
      m = flax_module.Module(f)
      rng = jax.random.PRNGKey(42)
      variables = m.init(rng)
      self.assertEqual(variables, {"params": {"~": {"w1": 0, "w2": 1}}})
  
      w1, w2 = m.apply(variables)
      self.assertIs(w1, variables["params"]["~"]["w1"])
      self.assertIs(w2, variables["params"]["~"]["w2"])
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import flax.linen
  from haiku._src import base
  from haiku._src import filtering
  from haiku._src import module
  from haiku._src import transform
  from haiku._src.flax import flax_module
  from haiku._src.flax import utils
  from haiku._src.nets import resnet
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling the FlaxModuleTest obj's underlying member method code with example usage: obj.test_transform() and returns: w1,  w2