code: def test_repr():
      class A(eqxi.Enumeration):
          x = "foo"
  
      assert str(A.x).endswith("test_enum.test_repr.<locals>.A<foo>")
  
      @jax.jit
      def run(a):
          assert str(a).endswith(
              "test_enum.test_repr.<locals>.A<traced>"
          )
  
      run(A.x)
  
file_path: equinox/tests/test_enum.py
imports: import equinox
  import equinox.internal
  import jax
  import jax.numpy
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 3 for calling a global function with example usage: test_repr() without return types