benchmarks.py

#!/usr/bin/env python3
from mupq import mupq
from interface import parse_arguments, get_platform

if __name__ == "__main__":
    args, rest = parse_arguments()
    platform, settings = get_platform(args)
    with platform:
        schemes = [s for s in rest if s not in ['--nostack',
                                                '--nospeed',
                                                '--nohashing',
                                                '--nosize']]
        # TODO: get this working
        #if "--nostack" not in rest:
        #    test = mupq.StackBenchmark(settings, platform)
        #    test.test_all(schemes)

        if "--nospeed" not in rest:
            test = mupq.SpeedBenchmark(settings, platform)
            test.test_all(schemes)

        if "--nohashing" not in rest:
            test = mupq.HashingBenchmark(settings, platform)
            test.test_all(schemes)

        # TODO: get this working
        #if "--nosize" not in rest:
        #    test = mupq.SizeBenchmark(settings, platform)
        #    test.test_all(schemes)