test_cpp.py

import os
import unittest

from utils import execute_remote, get_ips


@unittest.skipIf(os.name == "nt", reason="Do not support windows yet")
def test_tensorpipe_comm():
    base_dir = os.environ.get("DIST_DGL_TEST_CPP_BIN_DIR", ".")
    ip_config = os.environ.get("DIST_DGL_TEST_IP_CONFIG", "ip_config.txt")
    client_bin = os.path.join(base_dir, "rpc_client")
    server_bin = os.path.join(base_dir, "rpc_server")
    ips = get_ips(ip_config)
    num_machines = len(ips)
    procs = []
    for ip in ips:
        procs.append(
            execute_remote(server_bin + " " + str(num_machines) + " " + ip, ip)
        )
    for ip in ips:
        procs.append(execute_remote(client_bin + " " + ip_config, ip))
    for p in procs:
        p.join()
        assert p.exitcode == 0