code: def _assert_vars_equal(obj1, obj2, varnames):
      for varname in varnames:
          vars1 = getattr(obj1, varname)
          vars2 = getattr(obj2, varname)
          for a, b in _safe_zip(vars1, vars2):
              assert (
                  a.aval.strip_weak_type() == b.aval.strip_weak_type()
              )
  
file_path: equinox/tests/test_finalise_jaxpr.py
imports: from typing import cast
  import equinox.internal
  import jax
  import jax.core
  import jax.lax
  import jax.numpy
  from .helpers import tree_allclose
label_desc: Use image node_id 2 for calling a global function with example usage: _assert_vars_equal(obj1, obj2, varnames) without return types