#!/usr/bin/env python3
"""Verify CSV output is valid."""

import csv
import json

# Verify CSV is valid
with open('/root/.hermes/wizard-outputs/spark/output.csv', 'r') as f:
    reader = csv.DictReader(f)
    rows = list(reader)

assert len(rows) == 2, f"Expected 2 rows, got {len(rows)}"
assert 'name' in rows[0], "Missing 'name' column"
assert rows[0]['name'] == 'Alice Smith', f"Wrong name: {rows[0]['name']}"

# Verify the JSON source was correctly processed
with open('/root/.hermes/wizard-outputs/spark/sample_data.json', 'r') as f:
    source = json.load(f)

assert len(source) == 2, "Source JSON should have 2 records"

print("PASS: CSV is valid with correct data structure")
print(f"Records: {len(rows)}")
print(f"Columns: {sorted(rows[0].keys())}")