task9Test.py

# -*- coding: utf-8 -*-
__author__ = "Hamahmi"
"""
Please put this file in the same dir as your .g4 file after compiling it.
"""

import argparse
import os
import random
import sys

parser = argparse.ArgumentParser(description="Task 9 Test")
parser.add_argument(
    "--name",
    "-n",
    type=str,
    help="Your file name : T07_37_1234_John_Smith",
    required=True,
)
parser.add_argument(
    "--random",
    "-r",
    dest="rt",
    action="store_true",
    help="add -r or --random to get randomized test cases",
)
parser.add_argument(
    "--size",
    "-s",
    type=int,
    default=23,
    help="if you used the random option, you can set the number of test cases",
)
parser.add_argument(
    "--print",
    "-p",
    dest="print",
    action="store_true",
    help="add -p or --print to print the test cases",
)
args = parser.parse_args()


def generate_lisp_list(size):
    if size == 0:
        return "nil"
    random_number = random.randint(0, 9)
    size = size - 1
    return "(" + str(random_number) + " . " + generate_lisp_list(size) + ")"


def generate_random_lisp_list():
    return generate_lisp_list(random.randint(0, 23))


def get_readable_representation(lisp_list):
    lisp_list = lisp_list.replace("(", "").replace(")", "").replace(" ", "").split(".")
    if lisp_list == ["nil"]:
        return "()"
    output = "("
    for elem in lisp_list:
        if elem != "nil":
            output += elem + ", "
    output = output[:-2] + ")"
    return output


def generate_random_test_cases(size):
    output = dict()
    counter = 0
    while counter < (size):
        lisp_list = generate_random_lisp_list()
        if not output.__contains__(lisp_list):
            output[lisp_list] = get_readable_representation(lisp_list)
            counter += 1
    return output


if __name__ == "__main__":

    random.seed(23)

    if args.rt:
        test_cases = generate_random_test_cases(args.size)
    else:
        test_cases = {"nil": "()", "(1 . nil)": "(1)", "(1 . (2 . nil))": "(1, 2)"}

    if args.print:
        for case, expected_output in test_cases.items():
            print(case + " -> " + expected_output)

    wrong = dict()
    test_size = len(test_cases.keys())

    i = 0
    for case, expected_output in test_cases.items():
        with open("temp.txt", "w") as f:
            f.write(case)

        output = (
            os.popen("java org.antlr.v4.gui.TestRig " + args.name + " start temp.txt")
            .read()
            .replace(" ", "")
            .replace("\n", "")
            .replace("\r", "")
        )

        if output == "":
            print("Please compile first!")
            sys.exit()

        ex_out = expected_output.replace(" ", "").replace("\n", "").replace("\r", "")

        if output != ex_out:
            wrong[case] = (ex_out, output)
        i += 1
        print("{:.1f}".format(i / test_size * 100.0) + " %", end="\r")

    if(sys.platform.startswith("win")):
        os.system("del temp.txt")
    else:
        os.system("rm temp.txt")
    print()
    if wrong == {}:
        print("Passed all " + str(test_size) + " test cases :)")
    else:
        print(
            "Wrong answer in "
            + str(len(wrong.keys()))
            + " / "
            + str(test_size)
            + " :( !!"
        )
        for case, outputs in wrong.items():
            print(
                "In test case : "
                + case
                + " the output : "
                + outputs[0]
                + " was expected, but got : "
                + outputs[1],
                end="",
            )