test.py

#!/usr/bin/env python3
"""Build and run top_storage_tb with iverilog/vvp.

Uses IVERILOG and VVP env vars if set; otherwise defaults to iverilog/vvp on PATH.
"""
from __future__ import annotations

import os
import subprocess
from pathlib import Path
ROOT = Path(__file__).resolve().parent
OUT = ROOT / "user" / "sim" / "calc_flow_tb.vvp"
IVERILOG = os.environ.get("IVERILOG", "iverilog")
VVP = os.environ.get("VVP", "vvp")
SRC = ROOT / "user" / "src"
DISP = SRC / "display" 
SOURCES = [
    ROOT / "user" / "sim" / "calc_flow_tb.v",
    ROOT / "user" / "src" / "top_fast.v",
    ROOT / "user" / "src" / "top.sv",
    ROOT / "user" / "src" / "button_hub.v",
    ROOT / "user" / "src" / "debounce_btn.v",
    ROOT / "user" / "src" / "controller" / "main_controller.v",
    ROOT / "user" / "src" / "config.v",
    ROOT / "user" / "src" / "uart" / "uart_rx_fifo.v",
    ROOT / "user" / "src" / "uart" / "uart_rx.v",
    ROOT / "user" / "src" / "uart" / "fifo_buffer.v",
    ROOT / "user" / "src" / "uart" / "uart_tx_fifo.v",
    ROOT / "user" / "src" / "uart" / "uart_tx.v",
    ROOT / "user" / "src" / "input_matrix" / "input_matrix.v",
    ROOT / "user" / "src" / "storage" / "storage_controller.sv",
    ROOT / "user" / "src" / "storage" / "storage_FSM.v",
    ROOT / "user" / "src" / "storage" / "matrix_memory.v",
    ROOT / "user" / "src" / "display" / "matrix_display_controller.v",
    DISP / "seven_seg_mux.v",
    SRC  / "storage" /"display_all.v",
    SRC / "storage"/"input_disposer.sv",
    SRC / "storage"/"load_matrix.sv",
]


def run(cmd: list[str]) -> None:
    print("[cmd]", " ".join(cmd))
    subprocess.run(cmd, check=True)


def compile_tb() -> None:
    cmd = [
        IVERILOG,
        "-g2012",
        "-o",
        str(OUT),
    ] + [str(p) for p in SOURCES]
    run(cmd)


def run_tb() -> None:
    cmd = [VVP, str(OUT)]
    run(cmd)


def main() -> None:
    compile_tb()
    run_tb()


if __name__ == "__main__":
    main()