code: def test_lambda_layer(getkey):
      net = eqx.nn.Sequential(
          [
              eqx.nn.Identity(),
              eqx.nn.Lambda(jnn.relu),
          ]
      )
      x = jnp.array([[-1, -2, -3], [1, 2, 3]])
      output = net(x)
      assert output.shape == (2, 3)
      assert (output >= 0).all()
  
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 47 for calling a global function with example usage: test_lambda_layer(getkey) without return types