Function call

python/oneflow/framework/unittest.py:174–182

(conn=None, cmd=None, msg=None)

def call(conn=None, cmd=None, msg=None):
    cmd = "call/" + cmd
    print("[unittest]", f"[{cmd}]", msg)
    conn.send(cmd.encode())
    msg_ = ""
    if msg is not None:
        msg_ = msg
    conn.send(msg_.encode())
    return conn.recv().decode()


TestCase = unittest.TestCase