code: def test_lift_toplevel(self):
      def f():
          mod = transform_flax.lift(Child(), name="foo")
          mod()
  
      f = transform.transform_with_state(f)
  
      params, state = f.init(jax.random.PRNGKey(42))
      self.assertEqual(params, {"foo/~": {"w": 0}})
      self.assertEqual(
          state,
          {"foo/flim_state/~": {"s": 1}, "foo/flam_state/~": {"s": 2}},
      )
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import flax.errors
  import flax.linen
  from haiku._src import base
  from haiku._src import module
  from haiku._src import transform
  from haiku._src.flax import transform_flax
  import jax
  import jax.numpy
label_desc: Use image node_id 2 for calling the TransformFlaxTest obj's underlying member method code with example usage: obj.test_lift_toplevel() without return types