#!/usr/bin/env python3
"""Compare a search result with the default inference suffix."""
import argparse
import json
from pathlib import Path


def main():
    root = Path(__file__).resolve().parents[2]
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument('--result', type=Path, required=True)
    parser.add_argument('--default', type=Path, default=root / 'configs/suffix.json')
    args = parser.parse_args()
    result = json.loads(args.result.read_text())
    candidate = result.get('best_projected', result)
    expected = json.loads(args.default.read_text())
    for field in ('suffix_text', 'suffix_token_ids'):
        if field not in candidate or field not in expected:
            raise SystemExit(f'Missing required field: {field}')
        if candidate[field] != expected[field]:
            raise SystemExit(f'Suffix mismatch: {field}')
    print('MATCH: suffix text and token IDs equal the default inference suffix.')


if __name__ == '__main__':
    main()