code: def roundtrip_node_data(self, example):
      original = registry.flatten(example)[1]
      restored = pytree.PyTreeDef.make_from_node_data_and_children(
          registry, original.node_data(), original.children()
      )
      self.assertEqual(restored, original)
  
file_path: tensorflow/third_party/xla/xla/python/pytree_test.py
imports: import collections
  from absl.testing import absltest
  from xla.python import xla_client
label_desc: Use image node_id 5 for calling the PyTreeTest obj's underlying member method code with example usage: obj.roundtrip_node_data(example) without return types