code: def check_tennis_data(event_id, node_count, mode, edge_cnt):
      loader = TwitterTennisDatasetLoader(
          event_id, N=node_count, feature_mode=mode
      )
      dataset = loader.get_dataset()
      for epoch in range(3):
          i = 0
          for snapshot in dataset:
              if node_count == 1000:
                  assert snapshot.edge_index.shape == (2, edge_cnt[i])
                  assert snapshot.edge_attr.shape == (edge_cnt[i],)
              else:
                  assert snapshot.edge_index.shape[1] <= edge_cnt[i]
                  assert snapshot.edge_attr.shape[0] <= edge_cnt[i]
              if mode == "encoded":
                  assert snapshot.x.shape == (node_count, 16)
              elif mode == "diagonal":
                  assert snapshot.x.shape == (node_count, node_count)
              else:
                  assert snapshot.x.shape == (node_count, 2)
              assert snapshot.y.shape == (node_count,)
              i += 1
  
imports: import numpy
  import networkx
  from torch_geometric_temporal.signal import temporal_signal_split
  from torch_geometric_temporal.signal import StaticGraphTemporalSignal
  from torch_geometric_temporal.signal import DynamicGraphTemporalSignal
  from torch_geometric_temporal.signal import DynamicGraphStaticSignal
  from torch_geometric_temporal.signal import StaticHeteroGraphTemporalSignal
  from torch_geometric_temporal.signal import DynamicHeteroGraphTemporalSignal
  from torch_geometric_temporal.signal import DynamicHeteroGraphStaticSignal
  from torch_geometric_temporal.dataset import METRLADatasetLoader, PemsBayDatasetLoader
  from torch_geometric_temporal.dataset import ChickenpoxDatasetLoader, PedalMeDatasetLoader, WikiMathsDatasetLoader, EnglandCovidDatasetLoader
  from torch_geometric_temporal.dataset import TwitterTennisDatasetLoader, MontevideoBusDatasetLoader, MTMDatasetLoader
  from torch_geometric_temporal.dataset import WindmillOutputLargeDatasetLoader, WindmillOutputMediumDatasetLoader, WindmillOutputSmallDatasetLoader
label_desc: Use image node_id 36 for calling a global function with example usage: check_tennis_data(event_id, node_count, mode, edge_cnt) without return types