code: def test_sage(g):
      data_info = {"num_nodes": g.num_nodes(), "out_size": 7}
      node_feat = None
      edge_feat = g.edata["scalar_w"]
  
      # node embedding
      model = GraphSAGE(data_info, embed_size=10)
      model(g, node_feat)
      model(g, node_feat, edge_feat)
  
      # node feat
      data_info["in_size"] = g.ndata["h"].shape[-1]
      node_feat = g.ndata["h"]
      model = GraphSAGE(data_info, embed_size=-1)
      model(g, node_feat)
      model(g, node_feat, edge_feat)
  
file_path: dgl/tests/go/test_model.py
imports: import dgl
  import pytest
  import torch
  from utils.graph_cases import get_cases
  from dglgo.model import *
label_desc: Use image node_id 6 for calling a global function with example usage: test_sage(g) without return types