Интегрирован wesp в сайт

2026-07-17 12:57:18 +03:00
parent 5dfa06ddbe
commit 355c0ef9f1
883 changed files with 194576 additions and 177 deletions
@@ -0,0 +1,95 @@
+"""Unit-тесты app.services.recipe_calculator (кроме unloading — см. test_recipe_calculator_unloading)."""
+
+from __future__ import annotations
+
+import unittest
+
+from app.services.recipe_calculator import (
+    calculate_ingredients,
+    calculate_ingredients_from_dry_matter,
+    calculate_recipe,
+    calculate_totals,
+    round_to_step5,
+)
+
+
+class RecipeCalculatorTests(unittest.TestCase):
+    def test_round_to_step5(self) -> None:
+        self.assertEqual(round_to_step5(12), 10)
+        self.assertEqual(round_to_step5(13), 15)
+        self.assertEqual(round_to_step5(100), 100)
+
+    def test_calculate_ingredients_basic(self) -> None:
+        ingredients = [
+            {"weightPerHead": 2.0, "dryMatter": 50.0, "component_id": "c1"},
+            {"weightPerHead": 3.0, "dryMatter": 60.0, "component_id": "c2"},
+        ]
+        result = calculate_ingredients(ingredients, heads_count=10, trip_percent=100.0)
+        self.assertEqual(len(result), 2)
+        self.assertEqual(result[0]["totalWeight"], 20.0)
+        self.assertEqual(result[0]["tripWeight"], 20.0)
+        self.assertEqual(result[0]["dryMatterPerHead"], 1.0)
+        self.assertEqual(result[1]["totalWeight"], 30.0)
+
+    def test_calculate_ingredients_uses_component_dm_map(self) -> None:
+        ingredients = [{"weightPerHead": 1.0, "dryMatter": 0, "component_id": "c-map"}]
+        result = calculate_ingredients(
+            ingredients,
+            heads_count=5,
+            trip_percent=50.0,
+            component_dry_matter_map={"c-map": 80.0},
+        )
+        self.assertEqual(result[0]["dryMatterPerHead"], 0.8)
+        self.assertEqual(result[0]["tripWeight"], 2.5)
+
+    def test_calculate_totals_sums_ingredients(self) -> None:
+        calculated = [
+            {"totalWeight": 20.0, "tripWeight": 10.0, "dryMatterPerHead": 1.0, "weightPerHead": 2.0},
+            {"totalWeight": 30.0, "tripWeight": 15.0, "dryMatterPerHead": 1.5, "weightPerHead": 3.0},
+        ]
+        totals = calculate_totals(calculated, [])
+        self.assertEqual(totals["totalWeight"], 50.0)
+        self.assertEqual(totals["totalTripWeight"], 25.0)
+        self.assertEqual(totals["totalDryMatterPerHead"], 2.5)
+        self.assertEqual(totals["totalWeightPerHead"], 5.0)
+
+    def test_calculate_ingredients_from_dry_matter_inverse(self) -> None:
+        ingredients = [{"dryMatterPerHead": 1.0, "dryMatter": 50.0}]
+        result = calculate_ingredients_from_dry_matter(
+            ingredients, heads_count=10, trip_percent=100.0
+        )
+        self.assertEqual(result[0]["weightPerHead"], 2.0)
+        self.assertEqual(result[0]["totalWeight"], 20.0)
+        self.assertEqual(result[0]["tripWeight"], 20.0)
+
+    def test_calculate_ingredients_from_dry_matter_zero_dm_percent_zeros_weight(self) -> None:
+        ingredients = [{"dryMatterPerHead": 1.0, "dryMatter": 0}]
+        result = calculate_ingredients_from_dry_matter(
+            ingredients, heads_count=10, trip_percent=100.0
+        )
+        self.assertEqual(result[0]["weightPerHead"], 0.0)
+
+    def test_calculate_recipe_full_pipeline_weight_mode(self) -> None:
+        result = calculate_recipe(
+            ingredients=[{"weightPerHead": 2.0, "dryMatter": 50.0}],
+            heads_count=100,
+            trip_percent=100.0,
+            unloading_groups=[{"distributionType": "percent", "value": 100}],
+        )
+        self.assertEqual(result["totals"]["totalTripWeight"], 200.0)
+        self.assertEqual(result["unloadingGroups"][0]["calculatedWeight"], 200)
+        self.assertEqual(result["unloadingTotals"]["totalPercent"], 100.0)
+
+    def test_calculate_recipe_from_dry_matter_mode(self) -> None:
+        result = calculate_recipe(
+            ingredients=[{"dryMatterPerHead": 0.5, "dryMatter": 50.0}],
+            heads_count=20,
+            trip_percent=50.0,
+            calculate_from_dry_matter=True,
+        )
+        self.assertEqual(result["ingredients"][0]["weightPerHead"], 1.0)
+        self.assertEqual(result["totals"]["totalTripWeight"], 10.0)
+
+
+if __name__ == "__main__":
+    unittest.main()