code: def nest_assert_equal(a, b):
      tree.map_structure(np.testing.assert_array_equal, a, b)
  
file_path: acme/acme/jax/savers_test.py
imports: from unittest import mock
  from acme import core
  from acme.jax import savers
  from acme.testing import test_utils
  from acme.utils import paths
  import jax.numpy
  import numpy
  import tree
  from absl.testing import absltest
label_desc: Use image node_id 1 for calling a global function with example usage: nest_assert_equal(a, b) without return types