docstr: Tests unpad(pad) is identity with random graphs.
docstr_tok: ["Tests","unpad","(","pad",")","is","identity","with","random","graphs","."]
code: def test_pad_unpad_with_graphs_exact_padding(self):
      """Tests unpad(pad) is identity with random graphs."""
      g = _get_random_graph(
          include_globals=True,
          include_node_features=True,
          include_edge_features=True,
      )
      recovered_g = utils.unpad_with_graphs(
          utils.pad_with_graphs(
              g,
              n_node=g.n_node.sum() + 1,
              n_edge=g.n_edge.sum(),
              n_graph=g.n_node.shape[0] + 1,
          )
      )
  
      jax.tree_util.tree_map(np.testing.assert_allclose, recovered_g, g)
  
imports: import functools
  import os
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax.lib import xla_bridge
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  from jraph._src import utils
  import numpy
label_desc: Use image node_id 9 for calling the GraphTest obj's underlying member method code with example usage: obj.test_pad_unpad_with_graphs_exact_padding() without return types