TDD: edit extract_labels_situation_avalancheuse method: test passes

qloridant · May 2, 2023 · afabab0 · afabab0
1 parent f7dd3cf
commit afabab0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bera/utils/bulletin.py b/bera/utils/bulletin.py
@@ -346,8 +346,8 @@ def extract_labels_situation_avalancheuse(raw_text: str) -> set[Label]:
             labels.add(Label.NEIGE_FRAICHE)
         if "neige humide" in raw_text.lower() or "neige de printemps" in raw_text.lower():
             labels.add(Label.NEIGE_HUMIDE)
-        if "plaque de fond" in raw_text.lower() or \
-                not re.match("avalanche[\s-] de glissement", raw_text.lower()) is None:
+        if "plaque de fond" in raw_text.lower() or "plaques de fond" in raw_text.lower() or\
+                not re.match("avalanches? de glissement", raw_text.lower()) is None:
             labels.add(Label.AVALANCHE_GLISSEMENT)
         return labels