code: def setUp(self):
      super(DataUtilsTest, self).setUp()
      self._test_graph = jraph.GraphsTuple(
          nodes=np.broadcast_to(
              np.arange(10, dtype=np.float32)[:, None], (10, 10)
          ),
          edges=np.concatenate(
              (
                  np.broadcast_to(
                      np.arange(20, dtype=np.float32)[:, None], (20, 4)
                  ),
                  np.broadcast_to(
                      np.arange(20, dtype=np.float32)[:, None], (20, 4)
                  ),
              )
          ),
          receivers=np.concatenate((np.arange(20), np.arange(20))),
          senders=np.concatenate((np.arange(20), np.arange(20))),
          globals={"label": np.array([1], dtype=np.int32)},
          n_node=np.array([10], dtype=np.int32),
          n_edge=np.array([40], dtype=np.int32),
      )
      ogb_path = pathlib.Path(data_utils.__file__).parents[0]
      master_csv_path = pathlib.Path(
          ogb_path, "test_data", "master.csv"
      )
      split_path = pathlib.Path(ogb_path, "test_data", "train.csv.gz")
      data_path = master_csv_path.parents[0]
      self._reader = data_utils.DataReader(
          data_path=data_path,
          master_csv_path=master_csv_path,
          split_path=split_path,
      )
  
file_path: jraph/jraph/ogb_examples/data_utils_test.py
imports: import pathlib
  from absl.testing import absltest
  from absl.testing import parameterized
  import jraph
  from jraph.ogb_examples import data_utils
  import numpy
  import tree
label_desc: Use image node_id 1 for calling the DataUtilsTest obj's underlying member method code with example usage: obj.setUp() without return types