util.py

import inspect
import os
from typing import Type
from edg import *


def run_test_board(design: Type[Block]) -> None:
    from edg import compile_board_inplace

    compile_board_inplace(design)

    dirname = os.path.join(os.path.dirname(inspect.getfile(design)), design.__name__)
    ref_net_filenames = [f for f in os.listdir(dirname) if f.startswith(design.__name__) and f.endswith(".net.ref")]
    net_filenames = [f for f in os.listdir(dirname) if f.startswith(design.__name__) and f.endswith(".net")]

    assert net_filenames, "no netlist files found"
    assert set(f[:-4] for f in ref_net_filenames) == set(
        net_filenames
    ), f"reference netlist files {ref_net_filenames} do not match generated netlist files {net_filenames}"

    for net_filename in net_filenames:
        with open(os.path.join(dirname, net_filename), newline=None) as f:
            generated_netlist = f.read()

        with open(os.path.join(dirname, net_filename + ".ref"), newline=None) as f:
            reference_netlist = f.read()

        assert (
            generated_netlist == reference_netlist
        ), f"netlist differs from reference for {design.__name__}, if this is expected you may need to update the reference"