code: def test_linear_rank2(self):
      def f():
          return basic.Linear(output_size=2)(jnp.zeros((5, 6)))
  
      init_fn, apply_fn = transform.transform(f)
      params = init_fn(random.PRNGKey(428))
      self.assertEqual(params["linear"]["w"].shape, (6, 2))
      self.assertEqual(params["linear"]["b"].shape, (2,))
      self.assertEqual(apply_fn(params, None).shape, (5, 2))
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import base
  from haiku._src import basic
  from haiku._src import test_utils
  from haiku._src import transform
  import jax
  from jax import random
  import jax.numpy
  import numpy
label_desc: Use image node_id 2 for calling the LinearTest obj's underlying member method code with example usage: obj.test_linear_rank2() and returns: basic