code: def test_csrmask(idtype, dtype, A_nnz, B_nnz):
      a, A = _random_simple_graph(
          idtype, dtype, F.ctx(), 500, 600, A_nnz, "A", "B", "AB"
      )
      b, B = _random_simple_graph(
          idtype, dtype, F.ctx(), 500, 600, B_nnz, "A", "B", "AB"
      )
      C = dgl._sparse_ops._csrmask(A._graph, A.edata["w"], B._graph)
      B_row, B_col = B.edges(order="eid")
      B_row = F.asnumpy(B_row)
      B_col = F.asnumpy(B_col)
      c = F.tensor(a.todense()[B_row, B_col], dtype)
      assert F.allclose(C, c)
  
file_path: dgl/tests/python/common/test_sparse_ops-csr.py
imports: import backend
  import dgl
  import numpy
  import pytest
  import scipy.sparse
  from utils import parametrize_idtype
label_desc: Use image node_id 6 for calling a global function with example usage: test_csrmask(idtype, dtype, A_nnz, B_nnz) without return types