code: def __init__(self, graphs):
      self.graphs = graphs
      super().__init__()
  
file_path: spektral/tests/test_models/core.py
imports: import numpy
  import scipy.sparse
  import tensorflow
  from spektral.data import Dataset, Graph, loaders
label_desc: Use image node_id 1 to create a new TestDataset object from inherited base classes: Dataset with example: obj = TestDataset(graphs)