code: def test_compute_edge_score_sigmoid():
      edge_index = torch.tensor(
          [
              [0, 0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 5],
              [1, 2, 3, 0, 2, 3, 0, 1, 3, 0, 1, 2, 5, 4],
          ]
      )
      raw = torch.randn(edge_index.size(1))
      e = EdgePooling.compute_edge_score_sigmoid(raw, edge_index, 6)
      assert torch.all(e >= 0) and torch.all(e <= 1)
      assert torch.all(torch.argsort(raw) == torch.argsort(e))
  
      if is_full_test():
          jit = torch.jit.script(EdgePooling.compute_edge_score_sigmoid)
          assert torch.allclose(jit(raw, edge_index, 6), e)
  
imports: import torch
  from torch_geometric.nn import EdgePooling
  from torch_geometric.testing import is_full_test
  from torch_geometric.utils import scatter
label_desc: Use image node_id 3 for calling a global function with example usage: test_compute_edge_score_sigmoid() without return types