docstr: Tests unpadding of graph.
docstr_tok: ["Tests","unpadding","of","graph","."]
code: def test_unpad(self):
      """Tests unpadding of graph."""
      _, graphs_tuple = _get_list_and_batched_graph()
      unpadded_graphs_tuple = utils.unpad_with_graphs(graphs_tuple)
      expected_unpadded_graph = graph.GraphsTuple(
          n_node=jnp.array([1, 3, 1, 0]),
          n_edge=jnp.array([2, 5, 0, 0]),
          nodes=_make_nest(jnp.arange(10).reshape(5, 2)),
          edges=_make_nest(jnp.arange(21).reshape(7, 3)),
          globals=_make_nest(jnp.arange(8).reshape(4, 2)),
          senders=jnp.array([0, 0, 1, 1, 2, 3, 3]),
          receivers=jnp.array([0, 0, 2, 1, 3, 2, 1]),
      )
      jax.tree_util.tree_map(
          np.testing.assert_allclose,
          unpadded_graphs_tuple,
          expected_unpadded_graph,
      )
  
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 7 for calling the GraphTest obj's underlying member method code with example usage: obj.test_unpad() without return types