code: def assert_matches(flax_collection, haiku_collection):
      flax_collection = utils.flatten_flax_to_haiku(flax_collection)
      for mod_name, name, value in filtering.traverse(haiku_collection):
          # We expect equality (not close) because we are running the same
          # operations in the same order on the same data in both cases.
          np.testing.assert_array_equal(
              flax_collection[mod_name][name],
              value,
              err_msg=f"{mod_name}/{name}",
          )
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import flax.linen
  from haiku._src import base
  from haiku._src import filtering
  from haiku._src import module
  from haiku._src import transform
  from haiku._src.flax import flax_module
  from haiku._src.flax import utils
  from haiku._src.nets import resnet
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: assert_matches(flax_collection, haiku_collection) without return types