code: def test_zero_padding_values(self, wrapper):
      g = _get_random_graph(max_n_graph=1)
      with self.subTest("test_all_padded_features"):
          self._assert_values_for_graph(
              utils.pad_with_graphs(g, n_node=20, n_edge=20, n_graph=3),
              wrapper=wrapper,
          )
      with self.subTest("test_no_edge_features"):
          self._assert_values_for_graph(
              utils.pad_with_graphs(
                  g,
                  n_node=sum(g.n_node) + 1,
                  n_edge=sum(g.n_edge),
                  n_graph=3,
              ),
              wrapper=wrapper,
          )
      with self.subTest("test_no_extra_graph_features"):
          self._assert_values_for_graph(
              utils.pad_with_graphs(
                  g,
                  n_node=sum(g.n_node) + 1,
                  n_edge=sum(g.n_edge),
                  n_graph=2,
              ),
              wrapper=wrapper,
          )
  
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 2 for calling the ZeroOutTest obj's underlying member method code with example usage: obj.test_zero_padding_values(wrapper) without return types