93 lines
2.7 KiB
Python
93 lines
2.7 KiB
Python
"""Tests de l'agrégation globale des têtes de minifigs."""
|
|
|
|
import csv
|
|
from pathlib import Path
|
|
|
|
from lib.rebrickable.global_minifig_heads import (
|
|
aggregate_global_heads_by_year,
|
|
write_global_heads_by_year,
|
|
)
|
|
|
|
|
|
def write_csv(path: Path, headers: list[str], rows: list[list[str]]) -> None:
|
|
"""Écrit un CSV simple pour les besoins des tests."""
|
|
with path.open("w", newline="") as csv_file:
|
|
writer = csv.writer(csv_file)
|
|
writer.writerow(headers)
|
|
writer.writerows(rows)
|
|
|
|
|
|
def test_aggregate_global_heads_by_year(tmp_path: Path) -> None:
|
|
"""Construit un agrégat global par année."""
|
|
inventories = tmp_path / "inventories.csv"
|
|
inventory_parts = tmp_path / "inventory_parts.csv"
|
|
parts = tmp_path / "parts.csv"
|
|
colors = tmp_path / "colors.csv"
|
|
sets = tmp_path / "sets.csv"
|
|
destination = tmp_path / "global_heads.csv"
|
|
write_csv(
|
|
inventories,
|
|
["id", "version", "set_num"],
|
|
[
|
|
["1", "1", "1000-1"],
|
|
["2", "2", "1000-1"],
|
|
["3", "1", "2000-1"],
|
|
],
|
|
)
|
|
write_csv(
|
|
inventory_parts,
|
|
["inventory_id", "part_num", "color_id", "quantity", "is_spare", "img_url"],
|
|
[
|
|
["2", "3626b", "1", "2", "False", ""],
|
|
["3", "3626b", "2", "1", "False", ""],
|
|
["3", "3001", "1", "10", "False", ""],
|
|
],
|
|
)
|
|
write_csv(
|
|
parts,
|
|
["part_num", "name", "part_cat_id", "part_material"],
|
|
[
|
|
["3626b", "Minifig Head", "59", "Plastic"],
|
|
["3001", "Brick 2 x 4", "11", "Plastic"],
|
|
],
|
|
)
|
|
write_csv(
|
|
colors,
|
|
["id", "name", "rgb", "is_trans", "num_parts", "num_sets", "y1", "y2"],
|
|
[
|
|
["1", "Yellow", "FFFF00", "False", "0", "0", "0", "0"],
|
|
["2", "Light Flesh", "FFE1BD", "False", "0", "0", "0", "0"],
|
|
],
|
|
)
|
|
write_csv(
|
|
sets,
|
|
["set_num", "name", "year", "theme_id", "num_parts", "img_url"],
|
|
[
|
|
["1000-1", "Set A", "2020", "1", "0", ""],
|
|
["2000-1", "Set B", "2021", "1", "0", ""],
|
|
],
|
|
)
|
|
|
|
rows = aggregate_global_heads_by_year(inventories, inventory_parts, parts, colors, sets)
|
|
write_global_heads_by_year(destination, rows)
|
|
|
|
with destination.open() as csv_file:
|
|
written = list(csv.DictReader(csv_file))
|
|
|
|
assert written == [
|
|
{
|
|
"year": "2020",
|
|
"color_rgb": "FFFF00",
|
|
"is_translucent": "false",
|
|
"color_name": "Yellow",
|
|
"quantity": "2",
|
|
},
|
|
{
|
|
"year": "2021",
|
|
"color_rgb": "FFE1BD",
|
|
"is_translucent": "false",
|
|
"color_name": "Light Flesh",
|
|
"quantity": "1",
|
|
},
|
|
]
|