litex/misoclib/tools/litescope/example_designs/test/make.py

#!/usr/bin/env python3
import argparse
import importlib


def _get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("-b", "--bridge", default="uart", help="Bridge to use")
    parser.add_argument("--port", default="2", help="UART port")
    parser.add_argument("--baudrate", default=115200, help="UART baudrate")
    parser.add_argument("--busword", default=32, help="CSR busword")

    parser.add_argument("test", nargs="+", help="specify a test")

    return parser.parse_args()

if __name__ == "__main__":
    args = _get_args()
    if args.bridge == "uart":
        from misoclib.com.uart.software.wishbone import UARTWishboneBridgeDriver
        port = args.port if not args.port.isdigit() else int(args.port)
        wb = UARTWishboneBridgeDriver(port, args.baudrate, "./csr.csv", int(args.busword), debug=False)
        ValueError("Invalid bridge {}".format(args.bridge))

    def _import(name):
        return importlib.import_module(name)

    for test in args.test:
        t = _import(test)
        t.main(wb)