code: def _simple_test(add, extract=lambda x: x):
      for _ in range(5):
          a = Tensor.randn(10, 10)
          b = Tensor.randn(10, 10)
          c = add(a, b)
          np.testing.assert_allclose(
              extract(c).numpy(),
              a.numpy() + b.numpy(),
              atol=1e-4,
              rtol=1e-5,
          )
      assert_jit_cache_len(add, 1)
  
imports: import unittest, functools
  import numpy
  from test.helpers import assert_jit_cache_len
  from tinygrad.tensor import Tensor
  from tinygrad.jit import TinyJit
  from tinygrad.device import Device
  from tinygrad.helpers import CI
label_desc: Use image node_id 1 for calling a global function with example usage: _simple_test(add, extract) without return types