43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
"""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
|