code: def test_basic(self):
      module = nnx.LinearGeneral(2, 3, rngs=nnx.Rngs(0))
      y = module(jnp.ones((1, 2)))
  
      assert y.shape == (1, 3)
      assert module.kernel.shape == (2, 3)
      assert module.bias is not None
      assert module.bias.shape == (3,)
  
file_path: flax/flax/experimental/nnx/tests/nn/test_linear.py
imports: import jax
  import jax.numpy
  from absl.testing import parameterized
  from jax.lax import Precision
  from numpy.testing import assert_array_equal
  from flax import linen
  from flax.experimental import nnx
label_desc: Use image node_id 1 for calling the TestLinearGeneral obj's underlying member method code with example usage: obj.test_basic() without return types