#!/usr/bin/env python3
"""Verify the scratch_analysis.json file is valid and contains expected data."""
import json
import sys

with open("/root/.hermes/wizard-outputs/spark/scratch_analysis.json") as f:
    data = json.load(f)

# Verify required fields
assert "total_files" in data, "Missing total_files"
assert "total_bytes" in data, "Missing total_bytes"
assert "file_types" in data, "Missing file_types"
assert "files" in data, "Missing files array"

# Verify counts match
assert data["total_files"] == len(data["files"]), "File count mismatch"
assert data["total_files"] == 165, f"Expected 165 files, got {data['total_files']}"
assert 230000 <= data["total_bytes"] <= 250000, f"Total bytes out of range: {data['total_bytes']}"

print("VERIFICATION PASSED")
print(f"Total files: {data['total_files']}")
print(f"Total bytes: {data['total_bytes']}")
print(f"File types: {data['file_types']}")