code: def test_dimenet_modules():
      env = Envelope(exponent=5)
      x = torch.randn(10, 3)
      assert env(x).size() == (10, 3)  # Isotonic layer.
  
      bbl = BesselBasisLayer(5)
      x = torch.randn(10, 3)
      assert bbl(x).size() == (10, 3, 5)  # Non-isotonic layer.
  
      rl = ResidualLayer(128, torch.nn.functional.relu)
      x = torch.randn(128, 128)
      assert rl(x).size() == (128, 128)
  
imports: import pytest
  import torch
  import torch.nn.functional
  from torch_geometric.nn import DimeNet, DimeNetPlusPlus
  from torch_geometric.nn.models.dimenet import BesselBasisLayer, Envelope, ResidualLayer
  from torch_geometric.testing import is_full_test, withPackage
label_desc: Use image node_id 1 for calling a global function with example usage: test_dimenet_modules() without return types