Premiers éléments de l'étude

2025-12-01 21:57:05 +01:00
commit 6494948b3c
50 changed files with 2595 additions and 0 deletions
--- a/tests/test_parts_inventory.py
+++ b/tests/test_parts_inventory.py
@@ -0,0 +1,140 @@
+"""Tests de construction du fichier parts_filtered.csv."""
+
+import csv
+from pathlib import Path
+
+from lib.rebrickable.parts_inventory import write_parts_filtered
+
+
+def write_csv(path: Path, headers: list[str], rows: list[list[str]]) -> None:
+    """Écrit un CSV simple pour les besoins de tests."""
+    with path.open("w", newline="") as csv_file:
+        writer = csv.writer(csv_file)
+        writer.writerow(headers)
+        writer.writerows(rows)
+
+
+def test_write_parts_filtered(tmp_path: Path) -> None:
+    """Assemble les pièces par set avec la dernière version d'inventaire."""
+    sets_path = tmp_path / "sets_enriched.csv"
+    inventories_path = tmp_path / "inventories.csv"
+    inventory_parts_path = tmp_path / "inventory_parts.csv"
+    colors_path = tmp_path / "colors.csv"
+    inventory_minifigs_path = tmp_path / "inventory_minifigs.csv"
+    minifigs_path = tmp_path / "minifigs.csv"
+    destination_path = tmp_path / "parts_filtered.csv"
+
+    write_csv(
+        sets_path,
+        ["set_num", "set_id", "name", "num_parts"],
+        [
+            ["1234-1", "1234", "Sample Set A", "9"],
+            ["5678-1", "5678", "Sample Set B", "2"],
+        ],
+    )
+    write_csv(
+        inventories_path,
+        ["id", "version", "set_num"],
+        [
+            ["1", "1", "1234-1"],
+            ["2", "2", "1234-1"],
+            ["3", "1", "5678-1"],
+            ["4", "1", "fig-123"],
+        ],
+    )
+    write_csv(
+        inventory_parts_path,
+        ["inventory_id", "part_num", "color_id", "quantity", "is_spare", "img_url"],
+        [
+            ["2", "3001", "1", "4", "False", ""],
+            ["2", "3002", "2", "1", "True", ""],
+            ["3", "3003", "3", "2", "False", ""],
+            ["4", "mf-1", "2", "1", "False", ""],
+            ["4", "mf-2", "3", "2", "False", ""],
+        ],
+    )
+    write_csv(
+        inventory_minifigs_path,
+        ["inventory_id", "fig_num", "quantity"],
+        [
+            ["2", "fig-123", "1"],
+        ],
+    )
+    write_csv(
+        minifigs_path,
+        ["fig_num", "name", "num_parts", "img_url"],
+        [
+            ["fig-123", "Sample Minifig", "2", ""],
+        ],
+    )
+    write_csv(
+        colors_path,
+        ["id", "name", "rgb", "is_trans", "num_parts", "num_sets", "y1", "y2"],
+        [
+            ["1", "White", "FFFFFF", "False", "0", "0", "0", "0"],
+            ["2", "Black", "000000", "True", "0", "0", "0", "0"],
+            ["3", "Red", "FF0000", "False", "0", "0", "0", "0"],
+        ],
+    )
+
+    write_parts_filtered(
+        sets_path,
+        inventories_path,
+        inventory_parts_path,
+        colors_path,
+        inventory_minifigs_path,
+        minifigs_path,
+        destination_path,
+    )
+
+    with destination_path.open() as result_file:
+        reader = csv.DictReader(result_file)
+        rows = list(reader)
+
+    assert rows == [
+        {
+            "part_num": "3001",
+            "color_rgb": "FFFFFF",
+            "is_translucent": "false",
+            "set_num": "1234-1",
+            "set_id": "1234",
+            "quantity_in_set": "4",
+            "is_spare": "false",
+        },
+        {
+            "part_num": "3002",
+            "color_rgb": "000000",
+            "is_translucent": "true",
+            "set_num": "1234-1",
+            "set_id": "1234",
+            "quantity_in_set": "1",
+            "is_spare": "true",
+        },
+        {
+            "part_num": "mf-1",
+            "color_rgb": "000000",
+            "is_translucent": "true",
+            "set_num": "1234-1",
+            "set_id": "1234",
+            "quantity_in_set": "1",
+            "is_spare": "false",
+        },
+        {
+            "part_num": "mf-2",
+            "color_rgb": "FF0000",
+            "is_translucent": "false",
+            "set_num": "1234-1",
+            "set_id": "1234",
+            "quantity_in_set": "2",
+            "is_spare": "false",
+        },
+        {
+            "part_num": "3003",
+            "color_rgb": "FF0000",
+            "is_translucent": "false",
+            "set_num": "5678-1",
+            "set_id": "5678",
+            "quantity_in_set": "2",
+            "is_spare": "false",
+        },
+    ]