code: def testGroverAttentionHead(grover_batched_graph):
      from deepchem.models.torch_models.grover_layers import (
          GroverAttentionHead,
      )
  
      (
          f_atoms,
          f_bonds,
          a2b,
          b2a,
          b2revb,
          a2a,
          _,
          _,
          _,
      ) = grover_batched_graph.get_components()
      hidden_size = 165
      atom_messages = False
      layer = GroverAttentionHead(
          hidden_size,
          bias=True,
          depth=4,
          undirected=False,
          atom_messages=atom_messages,
      )
      query, key, value = layer(f_atoms, f_bonds, a2b, a2a, b2a, b2revb)
      assert query.size() == (f_bonds.shape[0], hidden_size)
      assert key.size() == (f_bonds.shape[0], hidden_size)
      assert value.size() == (f_bonds.shape[0], hidden_size)
  
file_path: deepchem/deepchem/models/torch_models/tests/test_grover_layers.py
imports: import pytest
label_desc: Use image node_id 6 for calling a global function with example usage: testGroverAttentionHead(grover_batched_graph) without return types