code: def roundtrip(self, example):
      original = registry.flatten(example)[1]
      self.assertEqual(
          pytree.PyTreeDef.deserialize_using_proto(
              registry, original.serialize_using_proto()
          ),
          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 1 for calling the PyTreeTest obj's underlying member method code with example usage: obj.roundtrip(example) without return types