QR-Code-generator/python/qrcodegen-batch-test.py

# 
# QR Code generator batch test (Python)
# 
# Runs various versions of the QR Code generator test worker as subprocesses,
# feeds each one the same random input, and compares their output for equality.
# 
# Copyright (c) Project Nayuki. (MIT License)
# https://www.nayuki.io/page/qr-code-generator-library
# 
# Permission is hereby granted, free of charge, to any person obtaining a copy of
# this software and associated documentation files (the "Software"), to deal in
# the Software without restriction, including without limitation the rights to
# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
# the Software, and to permit persons to whom the Software is furnished to do so,
# subject to the following conditions:
# - The above copyright notice and this permission notice shall be included in
#   all copies or substantial portions of the Software.
# - The Software is provided "as is", without warranty of any kind, express or
#   implied, including but not limited to the warranties of merchantability,
#   fitness for a particular purpose and noninfringement. In no event shall the
#   authors or copyright holders be liable for any claim, damages or other
#   liability, whether in an action of contract, tort or otherwise, arising from,
#   out of or in connection with the Software or the use or other dealings in the
#   Software.
# 

import itertools, random, subprocess, sys, time
from typing import List, Optional, TypeVar


CHILD_PROGRAMS: List[List[str]] = [
	["python3", "-B", "../python/qrcodegen-worker.py"],  # Python program
	["java", "-cp", "../java/src/main/java", "-ea:io.nayuki.qrcodegen...", "io/nayuki/qrcodegen/QrCodeGeneratorWorker"],  # Java program
	["node", "../typescript-javascript/qrcodegen-worker.js"],  # TypeScript program
	["../c/qrcodegen-worker"],  # C program
	["../cpp/QrCodeGeneratorWorker"],  # C++ program
	["../rust/target/debug/examples/qrcodegen-worker"],  # Rust program
]


subprocs: List[subprocess.Popen] = []

def main() -> None:
	# Launch workers
	global subprocs
	try:
		for args in CHILD_PROGRAMS:
			subprocs.append(subprocess.Popen(args, universal_newlines=True,
				stdin=subprocess.PIPE, stdout=subprocess.PIPE))
	except FileNotFoundError:
		write_all(-1)
		raise
	
	# Check if any died
	time.sleep(0.3)
	if any(proc.poll() is not None for proc in subprocs):
		for proc in subprocs:
			if proc.poll() is None:
				print(-1, file=proc.stdin)
				not_none(proc.stdin).flush()
		sys.exit("Error: One or more workers failed to start")
	
	# Do tests
	for i in itertools.count():
		print("Trial {}: ".format(i), end="")
		do_trial()
		print()


def do_trial() -> None:
	mode: int = random.randrange(4)
	length: int
	data: List[int]
	if mode == 0:  # Numeric
		length = round((2 * 7089) ** random.random())
		data = random.choices(b"0123456789", k=length)
	elif mode == 1:  # Alphanumeric
		length = round((2 * 4296) ** random.random())
		data = random.choices(b"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ $%*+-./:", k=length)
	elif mode == 2:  # ASCII
		length = round((2 * 2953) ** random.random())
		data = [random.randrange(128) for _ in range(length)]
	elif mode == 3:  # Byte
		length = round((2 * 2953) ** random.random())
		data = [random.randrange(256) for _ in range(length)]
	else:
		raise AssertionError()
	
	write_all(length)
	for b in data:
		write_all(b)
	
	errcorlvl: int = random.randrange(4)
	minversion: int = random.randint(1, 40)
	maxversion: int = random.randint(1, 40)
	if minversion > maxversion:
		minversion, maxversion = maxversion, minversion
	mask: int = -1
	if random.random() < 0.5:
		mask = random.randrange(8)
	boostecl = int(random.random() < 0.2)
	print("mode={} len={} ecl={} minv={} maxv={} mask={} boost={}".format(mode, length, errcorlvl, minversion, maxversion, mask, boostecl), end="")
	
	write_all(errcorlvl)
	write_all(minversion)
	write_all(maxversion)
	write_all(mask)
	write_all(boostecl)
	flush_all()
	
	version: int = read_verify()
	print(" version={}".format(version), end="")
	if version == -1:
		return
	size: int = version * 4 + 17
	for _ in range(size**2):
		read_verify()


def write_all(val: int) -> None:
	for proc in subprocs:
		print(val, file=proc.stdin)

def flush_all() -> None:
	for proc in subprocs:
		not_none(proc.stdin).flush()

def read_verify() -> int:
	val: str = not_none(subprocs[0].stdout).readline().rstrip("\r\n")
	for proc in subprocs[1 : ]:
		if not_none(proc.stdout).readline().rstrip("\r\n") != val:
			raise ValueError("Mismatch")
	return int(val)


T = TypeVar("T")
def not_none(obj: Optional[T]) -> T:
	if obj is None:
		raise TypeError()
	return obj


if __name__ == "__main__":
	main()