mirror of
https://github.com/circify/circ.git
synced 2026-05-14 03:00:33 -04:00
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from subprocess import Popen, PIPE
|
|
import sys
|
|
from typing import List
|
|
import time
|
|
|
|
def flatten_args(args: dict) -> list:
|
|
''' flatten dictionary into list '''
|
|
flat_args = []
|
|
for k, v in args.items():
|
|
flat_args.append(str(k))
|
|
flat_args.append(str(v))
|
|
return flat_args
|
|
|
|
def build_server_cmd(exec: str, args: dict) -> List[str]:
|
|
return [exec, "-r", "0", "-i"] + flatten_args(args)
|
|
|
|
def build_client_cmd(exec: str, args: dict) -> List[str]:
|
|
return [exec, "-r", "1", "-i"] + flatten_args(args)
|
|
|
|
def run_test(desc: str, expected: str, server_cmd: List[str], client_cmd: List[str]) -> bool:
|
|
assert len(server_cmd) > 3, "server cmd does not have enough arguments"
|
|
assert len(client_cmd) > 3, "client cmd does not have enough arguments"
|
|
|
|
assert server_cmd[0] == client_cmd[0], "server and client do not have the same cmd: " + server_cmd[0] + ", " + client_cmd[0]
|
|
|
|
try:
|
|
server_proc = Popen(server_cmd, stdout=PIPE, stderr=PIPE)
|
|
client_proc = Popen(client_cmd, stdout=PIPE, stderr=PIPE)
|
|
|
|
server_out, server_err = server_proc.communicate()
|
|
client_out, client_err = client_proc.communicate()
|
|
|
|
assert not server_err, "server cmd has an error"
|
|
assert not client_err, "client cmd has an error"
|
|
|
|
server_out = server_out.decode('utf-8').strip()
|
|
client_out = client_out.decode('utf-8').strip()
|
|
|
|
assert server_out == client_out, "server out != client out\nserver_out: "+server_out+"\nclient_out: "+client_out
|
|
assert server_out == expected, "output != expected\nserver_out: "+server_out+"\nexpected: "+expected
|
|
return True
|
|
except Exception as e:
|
|
# print("Exception: ", e)
|
|
return False
|
|
|
|
|
|
def init_progress_bar(toolbar_width=40):
|
|
print("Running ABY unit tests")
|
|
sys.stdout.write("[%s]" % (" " * toolbar_width))
|
|
sys.stdout.flush()
|
|
sys.stdout.write("\b" * (toolbar_width+1))
|
|
|
|
def update_progress_bar():
|
|
sys.stdout.write("=")
|
|
sys.stdout.flush()
|
|
|
|
def end_progress_bar():
|
|
sys.stdout.write("]\n") # this ends the progress bar |