richard/etude_lego_jurassic_world
1
Fork 0
Code Issues Activity

Collage des pièces imprimées exclusives et variante sans impressions

Browse Source
This commit is contained in:
Richard Dern
2025-12-03 17:37:01 +01:00
parent 8560f15b41
commit cebe3046db
6 changed files with 117 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_part_rarity.py
View File

@@ -117,6 +117,7 @@ def test_build_part_rarity_counts_spares_and_ignores_categories(tmp_path: Path)
"part_category": "Bricks",
"sample_set_num": "2000-1",
"sample_set_id": "2000",
"sample_set_year": "2021",
"filtered_quantity": "1",
"filtered_set_count": "1",
"other_sets_quantity": "0",
@@ -130,6 +131,7 @@ def test_build_part_rarity_counts_spares_and_ignores_categories(tmp_path: Path)
"part_category": "Bricks",
"sample_set_num": "1000-1",
"sample_set_id": "1000",
"sample_set_year": "2020",
"filtered_quantity": "3",
"filtered_set_count": "2",
"other_sets_quantity": "3",
@@ -143,6 +145,7 @@ def test_build_part_rarity_counts_spares_and_ignores_categories(tmp_path: Path)
"part_category": "Large Buildable Figures",
"sample_set_num": "2000-1",
"sample_set_id": "2000",
"sample_set_year": "2021",
"filtered_quantity": "2",
"filtered_set_count": "1",
"other_sets_quantity": "4",
@@ -175,6 +178,7 @@ def test_write_part_rarity_outputs_csv(tmp_path: Path) -> None:
"part_category": "Bricks",
"sample_set_num": "123-1",
"sample_set_id": "123",
"sample_set_year": "2020",
"filtered_quantity": "3",
"filtered_set_count": "2",
"other_sets_quantity": "3",
@@ -188,7 +192,7 @@ def test_write_part_rarity_outputs_csv(tmp_path: Path) -> None:
assert destination.exists()
content = destination.read_text().strip().splitlines()
assert content[0] == (
"part_num,part_name,part_cat_id,part_category,sample_set_num,sample_set_id,filtered_quantity,filtered_set_count,"
"part_num,part_name,part_cat_id,part_category,sample_set_num,sample_set_id,sample_set_year,filtered_quantity,filtered_set_count,"
"other_sets_quantity,catalog_total_quantity,filtered_share"
)
assert content[1] == "p1,Brick 1x1,1,Bricks,123-1,123,3,2,3,6,0.5000"
assert content[1] == "p1,Brick 1x1,1,Bricks,123-1,123,2020,3,2,3,6,0.5000"

33
tests/test_part_rarity_printed_collage.py Normal file
View File

@@ -0,0 +1,33 @@
"""Tests du collage des pièces imprimées exclusives."""
import matplotlib
from pathlib import Path
from PIL import Image
from lib.plots.part_rarity import plot_printed_exclusive_parts
matplotlib.use("Agg")
def test_plot_printed_exclusive_parts(tmp_path: Path) -> None:
"""Génère un collage des pièces imprimées exclusives avec images locales."""
data_path = tmp_path / "part_rarity.csv"
resources_dir = tmp_path / "figures" / "rebrickable"
resources_dir.mkdir(parents=True)
(resources_dir / "1000" / "rare_parts").mkdir(parents=True)
(resources_dir / "2000" / "rare_parts").mkdir(parents=True)
Image.new("RGB", (60, 40), color=(255, 0, 0)).save(resources_dir / "1000" / "rare_parts" / "p1.jpg")
Image.new("RGB", (60, 40), color=(0, 255, 0)).save(resources_dir / "2000" / "rare_parts" / "p2.jpg")
data_path.write_text(
"part_num,part_name,part_cat_id,part_category,sample_set_num,sample_set_id,sample_set_year,filtered_quantity,filtered_set_count,other_sets_quantity,catalog_total_quantity,filtered_share\n"
"p1,Slope print,1,Bricks,1000-1,1000,2020,3,2,0,3,1.0000\n"
"p2,Tile print,1,Bricks,2000-1,2000,2021,2,1,0,2,1.0000\n"
"p3,Tile plain,1,Bricks,2000-1,2000,2021,2,1,0,2,1.0000\n"
)
destination = tmp_path / "figures" / "step34" / "printed_exclusive_parts.png"
plot_printed_exclusive_parts(data_path, destination, resources_dir=resources_dir, columns=2)
assert destination.exists()
assert destination.stat().st_size > 0