code: def test_rng_stream(self):
      key0 = jax.random.key(0)
      rngs = nnx.Rngs(params=key0)
      assert rngs._rngs["params"].counts[-1] == 0
  
      key1 = rngs.params()
      assert rngs._rngs["params"].counts[-1] == 1
      assert rngs._rngs["params"].key is key0
      assert not jnp.allclose(key0, key1)
  
      key2 = rngs.params()
      assert rngs._rngs["params"].counts[-1] == 2
      assert rngs._rngs["params"].key is key0
      assert not jnp.allclose(key1, key2)
  
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 5 for calling the TestRngs obj's underlying member method code with example usage: obj.test_rng_stream() without return types