docstr: Tests batching of graph.
docstr_tok: ["Tests","batching","of","graph","."]
code: def test_batch(self):
      """Tests batching of graph."""
      (
          list_graphs_tuple,
          batched_graphs_tuple,
      ) = _get_list_and_batched_graph()
      graphs_tuple = utils.batch(list_graphs_tuple)
      jax.tree_util.tree_map(
          np.testing.assert_allclose, graphs_tuple, batched_graphs_tuple
      )
  
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 1 for calling the GraphTest obj's underlying member method code with example usage: obj.test_batch() without return types