code: def test_partition_merge(self):
      rngs = nnx.Rngs(dropout=0)
  
      keys = rngs.fork()
  
      assert "dropout" in keys
      assert keys["dropout"].counts == [0, 0]
  
      rngs2 = nnx.Rngs(keys)
  
      key1 = rngs.dropout()
      key2 = rngs2.dropout()
      assert not jnp.allclose(key1, key2)
  
      rngs3 = nnx.Rngs(keys)
      key3 = rngs3.dropout()
      assert jnp.allclose(key2, key3)
  
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 8 for calling the TestRngs obj's underlying member method code with example usage: obj.test_partition_merge() without return types