code: def test_fork_split(self):
      rngs = nnx.Rngs(params=0, dropout=1)
      keys = rngs.fork(4)  # split all
  
      assert keys["params"].key.shape == (4,)
      assert keys["dropout"].key.shape == (4,)
  
imports: from typing import Any
  import jax
  import jax.numpy
  import pytest
  from flax.experimental import nnx
  from flax.experimental.nnx.nnx.rnglib import _stable_hash
label_desc: Use image node_id 10 for calling the TestRngs obj's underlying member method code with example usage: obj.test_fork_split() without return types