"""Tests du graphique minifigs par personnage."""

import matplotlib
from pathlib import Path

from lib.plots.minifig_characters import plot_minifigs_per_character
from lib.plots.minifig_characters import plot_character_variations_vs_total


matplotlib.use("Agg")


def test_plot_minifigs_per_character(tmp_path: Path) -> None:
    """Génère l'image de comptage par personnage."""
    counts_path = tmp_path / "counts.csv"
    destination = tmp_path / "figures" / "step22" / "minifig_characters.png"
    counts_path.write_text(
        "known_character,gender,minifig_count\n"
        "Owen Grady,male,2\n"
        "Figurant,unknown,1\n"
    )

    plot_minifigs_per_character(counts_path, destination)

    assert destination.exists()
    assert destination.stat().st_size > 0


def test_plot_character_variations_vs_total(tmp_path: Path) -> None:
    """Génère l'image comparant total et variations par personnage."""
    counts_path = tmp_path / "variations.csv"
    destination = tmp_path / "figures" / "step22" / "minifig_character_variations_totals.png"
    counts_path.write_text(
        "known_character,gender,variation_count,total_minifigs\n"
        "Owen Grady,male,2,3\n"
        "Ellie Sattler,female,1,2\n"
    )

    plot_character_variations_vs_total(counts_path, destination)

    assert destination.exists()
    assert destination.stat().st_size > 0