benchmark_app.py

import argparse

from src.benchmark import Benchmark


def main(args):
    mode = args.mode
    iterations = args.number_amount
    bits = args.bits

    if mode not in range(3):
        raise Exception('Mode ' + str(mode) + ' not allowed.')

    if iterations < 0:
        raise Exception('Amount ' +  str(iterations) + ' not allowed.')

    if bits < 0:
        raise Exception('Number of bits ' + str(bits) + ' not allowed.')

    benchmark = Benchmark(mode, iterations, bits)
    results = (benchmark.execute())

    print(results)


def print_help():
    print("""-m --mode : Simulation mode: 0 for local | 1 for IBM server
                simulation | 2 for IBM server REAL experiment\n\n"""
          + """-n --number_amount : Amount of numbers to generate.
          Must be greater than 0\n\n"""
          + """"-b --bits : Generates a random number between 0 and 2**b-1,
          i.e., a number with b binary bits""")


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '-m',
        '--mode',
        help="""Simulation mode: 0 for local | 1 for IBM server
                simulation | 2 for IBM server REAL experiment""",
        default=0,
        type=int)
    parser.add_argument(
        '-n',
        '--number_amount',
        help="Amount of numbers to generate. Needs to be greater than 0",
        default=1,
        type=int)
    parser.add_argument(
        '-b',
        '--bits',
        help="""Generates numbers of b bits""",
        default=1,
        type=int)

    parsed_args = parser.parse_args()
    main(parsed_args)