Write tests for bcae8 evaluator

MTES-MCT · Nov 28, 2024 · a9c74eb · a9c74eb
1 parent fec6507
commit a9c74eb
Show file tree

Hide file tree

Showing 3 changed files with 237 additions and 92 deletions.
diff --git a/envergo/hedges/models.py b/envergo/hedges/models.py
@@ -78,3 +78,17 @@ def hedges_to_remove(self):
 
     def length_to_remove(self):
         return round(sum(h.length for h in self.hedges_to_remove()))
+
+    def lineaire_detruit_pac(self):
+        return sum(
+            h.length
+            for h in self.hedges_to_remove()
+            if h.is_on_pac and h.hedge_type != "alignement"
+        )
+
+    def lineaire_type_4_sur_parcelle_pac(self):
+        return sum(
+            h.length
+            for h in self.hedges_to_remove()
+            if h.is_on_pac and h.hedge_type == "alignement"
+        )
diff --git a/envergo/moulinette/regulations/conditionnalitepac.py b/envergo/moulinette/regulations/conditionnalitepac.py
@@ -151,9 +151,7 @@ def get_catalog_data(self):
         lineaire_type_4_sur_parcelle_pac = 0
         if "lineaire_total" in self.catalog:
             haies = self.catalog["haies"]
-            for haie in haies:
-                if haie.is_on_pac and haie.hedge_type == "alignement":
-                    lineaire_type_4_sur_parcelle_pac += haie.length
+            lineaire_type_4_sur_parcelle_pac = haies.lineaire_type_4_sur_parcelle_pac()
         catalog["lineaire_type_4_sur_parcelle_pac"] = lineaire_type_4_sur_parcelle_pac
 
         return catalog
@@ -162,19 +160,12 @@ def get_result_data(self):
         is_small = False
         haies = self.catalog["haies"]
         lineaire_detruit_pac = 0
-        lineaire_type_4_sur_parcelle_pac = 0
         ratio_detruit = 0
         if "lineaire_total" in self.catalog:
             lineaire_total = self.catalog["lineaire_total"]
-            for haie in haies:
-                if haie.is_on_pac and haie.hedge_type != "alignement":
-                    lineaire_detruit_pac += haie.length
-
-                if haie.is_on_pac and haie.hedge_type == "alignement":
-                    lineaire_type_4_sur_parcelle_pac += haie.length
-
+            lineaire_detruit_pac = haies.lineaire_detruit_pac()
             ratio_detruit = lineaire_detruit_pac / lineaire_total
-            is_small = ratio_detruit <= 2 or lineaire_detruit_pac <= 5
+            is_small = ratio_detruit <= 0.02 or lineaire_detruit_pac <= 5
 
         return (
             # Main vars