code: def test_schnet_batch():
      num_graphs = 3
      batch = [generate_data() for _ in range(num_graphs)]
      batch = Batch.from_data_list(batch)
  
      model = SchNet(
          hidden_channels=16,
          num_filters=16,
          num_interactions=2,
          num_gaussians=10,
          cutoff=6.0,
      )
  
      with torch.no_grad():
          out = model(batch.z, batch.pos, batch.batch)
          assert out.size() == (num_graphs, 1)
  
file_path: pytorch_geometric/test/nn/models/test_schnet.py
imports: import pytest
  import torch
  from torch_geometric.data import Batch, Data
  from torch_geometric.nn import SchNet
  from torch_geometric.nn.models.schnet import RadiusInteractionGraph
  from torch_geometric.testing import is_full_test, withPackage
label_desc: Use image node_id 3 for calling a global function with example usage: test_schnet_batch() without return types