code: def test_frozen_dict_partially_maps(self):
      x = jax.tree_util.tree_map(
          lambda a, b: (a, b), freeze({"a": 2}), freeze({"a": {"b": 1}})
      )
      self.assertEqual(unfreeze(x), {"a": (2, {"b": 1})})
  
file_path: flax/tests/core/core_frozen_dict_test.py
imports: import jax
  from absl.testing import absltest, parameterized
  from flax.core import FrozenDict, copy, freeze, pop, pretty_repr, unfreeze
label_desc: Use image node_id 4 for calling the FrozenDictTest obj's underlying member method code with example usage: obj.test_frozen_dict_partially_maps() without return types