code: def check_binary_op(key1, key2, key3, op):
      for i in range(0, dist_g.num_edges(), 1000):
          i_end = min(i + 1000, dist_g.num_edges())
          assert F.array_equal(
              dist_g.edata[key3][i:i_end],
              op(
                  dist_g.edata[key1][i:i_end],
                  dist_g.edata[key2][i:i_end],
              ),
          )
  
file_path: dgl/tests/distributed/test_dist_tensor.py
imports: import operator
  import os
  import unittest
  import backend
  import dgl
  import pytest
  from utils import create_random_graph, generate_ip_config, reset_envs
label_desc: Use image node_id 3 for calling a global function with example usage: check_binary_op(key1, key2, key3, op) without return types