docstr: Tests construction of graph padding mask.
docstr_tok: ["Tests","construction","of","graph","padding","mask","."]
code: def test_get_graph_padding_mask(self):
      """Tests construction of graph padding mask."""
      _, graphs_tuple = _get_list_and_batched_graph()
      expected_mask = jnp.array([1, 1, 1, 1, 0, 0, 0]).astype(bool)
      with self.subTest("nojit"):
          mask = utils.get_graph_padding_mask(graphs_tuple)
          np.testing.assert_array_equal(mask, expected_mask)
      with self.subTest("jit"):
          mask = jax.jit(utils.get_graph_padding_mask)(graphs_tuple)
          np.testing.assert_array_equal(mask, expected_mask)
  
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 15 for calling the GraphTest obj's underlying member method code with example usage: obj.test_get_graph_padding_mask() without return types