code: def test_different_sequence_types(self, type_of_sequence):
      f_map = FlatMap(
          {
              "foo": type_of_sequence((1, 2)),
              "bar": type_of_sequence((3, {"b": 4})),
          }
      )
      leaves, _ = jax.tree_util.tree_flatten(f_map)
  
      self.assertEqual(leaves, [3, 4, 1, 2])
      self.assertEqual(f_map["foo"][0], 1)
      self.assertEqual(f_map["bar"][1]["b"], 4)
  
imports: import collections
  import copy
  import pickle
  import threading
  from absl.testing import absltest
  from absl.testing import parameterized
  import cloudpickle
  import dill
  from haiku._src import data_structures
  import jax
  import tree
label_desc: Use image node_id 27 for calling the FlatMappingTest obj's underlying member method code with example usage: obj.test_different_sequence_types(type_of_sequence) without return types