test.py

from pathlib import Path
import dsatpy
import json, time

def cpp_json_to_bitset_cnf(
    fpath: Path
) -> tuple[list[int], list[tuple[tuple[int, dsatpy.BitSet], ...]]]:
    with open(fpath, 'r') as f:
        data = json.load(f)
    vs = [card for _, card in data["vars"]]
    cnf = []
    for c in data["clauses"]:
        cnf.append([(var_idx - 1,
                     dsatpy.BitSet.from_decimal_string(str(decimal_str)))
                    for var_idx, decimal_str in c])
    return vs, cnf

def tmp_run_dsat(
        discrete_vars: list[int],
        cnf: list[list[tuple[int, dsatpy.BitSet]]],
        solver: dsatpy.Solver | None = None):

    if solver is None:
        solver = dsatpy.Solver()
    tic = time.perf_counter()

    solver.init(discrete_vars, cnf)
    is_sat = solver.solve()
    toc = time.perf_counter()
    runtime = toc - tic
    avg_strength = solver.t_strength / (solver.n_added_cdc + 1)
    return is_sat, runtime

def dsat_cpp_wrapper(cnf_path: Path) :
    vs, clauses = cpp_json_to_bitset_cnf(cnf_path)
    return tmp_run_dsat(vs, clauses)

if __name__ == "__main__":
    import sys
    results = dsat_cpp_wrapper(Path(sys.argv[1]))
    print(results)