code: def test_reader_repeat(self):
      self._reader.repeat()
      next(self._reader)
      graph = next(self._reader)
      # One graph in the test dataset so should be the same.
      tree.map_structure(
          np.testing.assert_almost_equal, graph, self._test_graph
      )
  
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 4 for calling the DataUtilsTest obj's underlying member method code with example usage: obj.test_reader_repeat() without return types