Momento/.claude/skills/suno-lyric-transformer/scripts/tests/test-assemble-summary.py

#!/usr/bin/env python3
# /// script
# requires-python = ">=3.10"
# dependencies = ["pytest>=7.0"]
# ///
"""Tests for assemble-summary.py"""

import json
import subprocess
import sys
from pathlib import Path

SCRIPT = str(Path(__file__).parent.parent / "assemble-summary.py")


def run_script(*args, input_data=None):
    """Run the script and return stdout and returncode."""
    result = subprocess.run(
        [sys.executable, SCRIPT, *args],
        capture_output=True, text=True,
        input=input_data
    )
    return result.stdout, result.returncode


def create_test_files(tmp_path):
    """Create sample JSON input files for testing."""
    validation = {
        "script": "validate-lyrics",
        "status": "pass",
        "metrics": {
            "total_lines": 20,
            "lyric_lines": 14,
            "section_count": 4,
            "sections": ["Verse 1", "Chorus", "Verse 2", "Chorus"]
        },
        "findings": [],
        "summary": {"total": 0}
    }

    syllables = {
        "script": "syllable-counter",
        "status": "pass",
        "metrics": {
            "total_lyric_lines": 14,
            "total_syllables": 112,
            "average_syllables_per_line": 8.0,
            "min_syllables": 5,
            "max_syllables": 12
        },
        "findings": [],
        "summary": {"total": 0}
    }

    cliches = {
        "script": "cliche-detector",
        "status": "pass",
        "metrics": {
            "total_cliches_found": 2,
            "categories": {"emotional": 1, "nature": 1}
        },
        "findings": [],
        "summary": {"total": 2}
    }

    val_file = tmp_path / "validation.json"
    syl_file = tmp_path / "syllables.json"
    cli_file = tmp_path / "cliches.json"

    val_file.write_text(json.dumps(validation))
    syl_file.write_text(json.dumps(syllables))
    cli_file.write_text(json.dumps(cliches))

    return str(val_file), str(syl_file), str(cli_file)


class TestAssembleSummary:
    def test_basic_assembly(self, tmp_path):
        val, syl, cli = create_test_files(tmp_path)
        output, code = run_script("--validation", val, "--syllables", syl, "--cliches", cli)
        assert code == 0
        assert "Transformation Summary" in output
        assert "Validation Status" in output
        assert "Sections:" in output

    def test_with_transformations(self, tmp_path):
        val, syl, cli = create_test_files(tmp_path)
        output, code = run_script(
            "--validation", val, "--syllables", syl, "--cliches", cli,
            "--transformations", "ST,CC,RA"
        )
        assert code == 0
        assert "Transformations Applied" in output
        assert "ST:" in output
        assert "CC:" in output
        assert "RA:" in output

    def test_json_output(self, tmp_path):
        val, syl, cli = create_test_files(tmp_path)
        out_file = tmp_path / "output.json"
        output, code = run_script(
            "--validation", val, "--syllables", syl, "--cliches", cli,
            "-o", str(out_file)
        )
        assert code == 0
        report = json.loads(out_file.read_text())
        assert report["script"] == "assemble-summary"
        assert "metrics" in report
        assert "markdown" in report

    def test_markdown_output_file(self, tmp_path):
        val, syl, cli = create_test_files(tmp_path)
        out_file = tmp_path / "output.md"
        output, code = run_script(
            "--validation", val, "--syllables", syl, "--cliches", cli,
            "-o", str(out_file)
        )
        assert code == 0
        content = out_file.read_text()
        assert "## Transformation Summary" in content

    def test_cliche_categories_displayed(self, tmp_path):
        val, syl, cli = create_test_files(tmp_path)
        output, code = run_script("--validation", val, "--syllables", syl, "--cliches", cli)
        assert code == 0
        assert "2 found" in output
        assert "emotional" in output
        assert "nature" in output

    def test_syllable_range_displayed(self, tmp_path):
        val, syl, cli = create_test_files(tmp_path)
        output, code = run_script("--validation", val, "--syllables", syl, "--cliches", cli)
        assert code == 0
        assert "5-12" in output
        assert "avg 8.0" in output

    def test_estimated_duration(self, tmp_path):
        val, syl, cli = create_test_files(tmp_path)
        output, code = run_script("--validation", val, "--syllables", syl, "--cliches", cli)
        assert code == 0
        # 4 sections * 15 sec = 60 sec = 1:00
        assert "1:00" in output

    def test_missing_files_handled(self, tmp_path):
        missing = str(tmp_path / "nonexistent.json")
        output, code = run_script(
            "--validation", missing, "--syllables", missing, "--cliches", missing
        )
        assert code == 2

    def test_help_flag(self):
        result = subprocess.run(
            [sys.executable, SCRIPT, "--help"],
            capture_output=True, text=True
        )
        assert result.returncode == 0
        assert "assemble" in result.stdout.lower()


if __name__ == "__main__":
    import pytest
    pytest.main([__file__, "-v"])