code: def test_dropout_inference(getkey):
      dropout = eqx.nn.Dropout()
      x = jrandom.normal(getkey(), (3, 4, 5))
      z1 = dropout(x, key=getkey(), inference=True)
      z2 = dropout(x, inference=True)
      assert jnp.all(x == z1)
      assert jnp.all(x == z2)
  
      dropout2 = eqx.nn.Dropout(inference=True)
      assert jnp.all(x == dropout2(x))
  
      dropout3 = eqx.tree_at(
          lambda d: d.inference, dropout2, replace=False
      )
      assert jnp.any(x != dropout3(x, key=jrandom.PRNGKey(0)))
  
imports: import warnings
  from typing import Union
  import equinox
  import jax
  import jax.nn
  import jax.numpy
  import jax.random
  import pytest
label_desc: Use image node_id 5 for calling a global function with example usage: test_dropout_inference(getkey) without return types