code: def _get_list_and_batched_graph():
      """Returns a list of individual graphs and a batched version.
  
      This test-case includes the following corner-cases:
        - single node,
        - multiple nodes,
        - no edges,
        - single edge,
        - and multiple edges.
      """
      batched_graph = graph.GraphsTuple(
          n_node=jnp.array([1, 3, 1, 0, 2, 0, 0]),
          n_edge=jnp.array([2, 5, 0, 0, 1, 0, 0]),
          nodes=_make_nest(jnp.arange(14).reshape(7, 2)),
          edges=_make_nest(jnp.arange(24).reshape(8, 3)),
          globals=_make_nest(jnp.arange(14).reshape(7, 2)),
          senders=jnp.array([0, 0, 1, 1, 2, 3, 3, 6]),
          receivers=jnp.array([0, 0, 2, 1, 3, 2, 1, 5]),
      )
  
      list_graphs = [
          graph.GraphsTuple(
              n_node=jnp.array([1]),
              n_edge=jnp.array([2]),
              nodes=_make_nest(jnp.array([[0, 1]])),
              edges=_make_nest(jnp.array([[0, 1, 2], [3, 4, 5]])),
              globals=_make_nest(jnp.array([[0, 1]])),
              senders=jnp.array([0, 0]),
              receivers=jnp.array([0, 0]),
          ),
          graph.GraphsTuple(
              n_node=jnp.array([3]),
              n_edge=jnp.array([5]),
              nodes=_make_nest(jnp.array([[2, 3], [4, 5], [6, 7]])),
              edges=_make_nest(
                  jnp.array(
                      [
                          [6, 7, 8],
                          [9, 10, 11],
                          [12, 13, 14],
                          [15, 16, 17],
                          [18, 19, 20],
                      ]
                  )
              ),
              globals=_make_nest(jnp.array([[2, 3]])),
              senders=jnp.array([0, 0, 1, 2, 2]),
              receivers=jnp.array([1, 0, 2, 1, 0]),
          ),
          graph.GraphsTuple(
              n_node=jnp.array([1]),
              n_edge=jnp.array([0]),
              nodes=_make_nest(jnp.array([[8, 9]])),
              edges=_make_nest(jnp.zeros((0, 3))),
              globals=_make_nest(jnp.array([[4, 5]])),
              senders=jnp.array([]),
              receivers=jnp.array([]),
          ),
          graph.GraphsTuple(
              n_node=jnp.array([0]),
              n_edge=jnp.array([0]),
              nodes=_make_nest(jnp.zeros((0, 2))),
              edges=_make_nest(jnp.zeros((0, 3))),
              globals=_make_nest(jnp.array([[6, 7]])),
              senders=jnp.array([]),
              receivers=jnp.array([]),
          ),
          graph.GraphsTuple(
              n_node=jnp.array([2]),
              n_edge=jnp.array([1]),
              nodes=_make_nest(jnp.array([[10, 11], [12, 13]])),
              edges=_make_nest(jnp.array([[21, 22, 23]])),
              globals=_make_nest(jnp.array([[8, 9]])),
              senders=jnp.array([1]),
              receivers=jnp.array([0]),
          ),
          graph.GraphsTuple(
              n_node=jnp.array([0]),
              n_edge=jnp.array([0]),
              nodes=_make_nest(jnp.zeros((0, 2))),
              edges=_make_nest(jnp.zeros((0, 3))),
              globals=_make_nest(jnp.array([[10, 11]])),
              senders=jnp.array([]),
              receivers=jnp.array([]),
          ),
          graph.GraphsTuple(
              n_node=jnp.array([0]),
              n_edge=jnp.array([0]),
              nodes=_make_nest(jnp.zeros((0, 2))),
              edges=_make_nest(jnp.zeros((0, 3))),
              globals=_make_nest(jnp.array([[12, 13]])),
              senders=jnp.array([]),
              receivers=jnp.array([]),
          ),
          graph.GraphsTuple(
              n_node=jnp.array([]),
              n_edge=jnp.array([]),
              nodes=_make_nest(jnp.zeros((0, 2))),
              edges=_make_nest(jnp.zeros((0, 3))),
              globals=_make_nest(jnp.zeros((0, 2))),
              senders=jnp.array([]),
              receivers=jnp.array([]),
          ),
      ]
  
      return list_graphs, batched_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 3 for calling a global function with example usage: _get_list_and_batched_graph() and returns: list_graphs,  batched_graph