code: def test_lstm_cell(getkey):
      gru = eqx.nn.LSTMCell(2, 8, key=getkey())
      h = jrandom.normal(getkey(), (8,)), jrandom.normal(getkey(), (8,))
      x = jrandom.normal(getkey(), (5, 2))
      for xi in x:
          h = gru(xi, h)
          h_, c_ = h
          assert h_.shape == (8,)
          assert c_.shape == (8,)
  
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 8 for calling a global function with example usage: test_lstm_cell(getkey) without return types