code: def test_integer_access(self):
      class Foo(nnx.Module):
          def __init__(self, *, rngs: nnx.Rngs):
              self.layers = [
                  nnx.Linear(1, 2, rngs=rngs),
                  nnx.Linear(2, 3, rngs=rngs),
              ]
  
      module = Foo(rngs=nnx.Rngs(0))
      state = module.get_state()
  
      assert module.layers[0].kernel.shape == (1, 2)
      assert state.layers[0].kernel.shape == (1, 2)
      assert module.layers[1].kernel.shape == (2, 3)
      assert state.layers[1].kernel.shape == (2, 3)
  
file_path: flax/flax/experimental/nnx/tests/test_state.py
imports: from absl.testing.absltest import TestCase
  from flax.experimental import nnx
label_desc: Use image node_id 5 for calling the StateTest obj's underlying member method code with example usage: obj.test_integer_access() without return types