hestia-earth-models 0.61.8__py3-none-any.whl → 0.62.1__py3-none-any.whl

tests/models/site/test_management.py

@@ -5,6 +5,7 @@ import pytest
 from hestia_earth.schema import TermTermType
 
 from hestia_earth.models.site.management import MODEL, MODEL_KEY, run, _should_run
+from hestia_earth.models.utils.blank_node import condense_nodes
 from tests.utils import fixtures_path
 
 CLASS_PATH = f"hestia_earth.models.{MODEL}.{MODEL_KEY}"
@@ -14,7 +15,8 @@ TERM_BY_ID = {
     'genericCropPlant': {'@type': 'Term', '@id': 'genericCropPlant', 'termType': TermTermType.LANDCOVER.value},
     'wheatPlant': {'@type': 'Term', '@id': 'wheatPlant', 'termType': TermTermType.LANDCOVER.value},
     'oatPlant': {'@type': 'Term', '@id': 'oatPlant', 'termType': TermTermType.LANDCOVER.value},
-    'agatiTree': {'@type': 'Term', '@id': 'agatiTree', 'termType': TermTermType.LANDCOVER.value}
+    'agatiTree': {'@type': 'Term', '@id': 'agatiTree', 'termType': TermTermType.LANDCOVER.value},
+    'wildGarlicPlant': {'@type': 'Term', '@id': 'wildGarlicPlant', 'termType': TermTermType.LANDCOVER.value},
 }
 
 
@@ -28,6 +30,152 @@ def lookup_side_effect(*args, **kwargs):
     return True
 
 
+@pytest.mark.parametrize(
+    "test_name,input_nodes,expected_output_nodes",
+    [
+        (
+            "No match",
+            [
+                {"startDate": "2001", "endDate": "2002", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+                {"startDate": "2003", "endDate": "2004", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [2]},
+            ],
+            [
+                {"startDate": "2001", "endDate": "2002", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+                {"startDate": "2003", "endDate": "2004", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [2]},
+            ],
+        ),
+        (
+            "No continuity",
+            [
+                {"startDate": "2001", "endDate": "2002", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+                {"startDate": "2004", "endDate": "2005", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+            ],
+            [
+                {"startDate": "2001", "endDate": "2002", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+                {"startDate": "2004", "endDate": "2005", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+            ],
+        ),
+        (
+            "No continuity (multiple values differ)",
+            [
+                {"startDate": "2001", "endDate": "2002", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [10, 20]},
+                {"startDate": "2003", "endDate": "2004", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [10, 30]},
+            ],
+            [
+                {"startDate": "2001", "endDate": "2002", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [10, 20]},
+                {"startDate": "2003", "endDate": "2004", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [10, 30]},
+            ],
+        ),
+        (
+            "2->1 condense (YYYY dates)",
+            [
+                {"startDate": "2001", "endDate": "2001", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [10, 20]},
+                {"startDate": "2002", "endDate": "2002", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [10, 20]},
+            ],
+            [
+                {"startDate": "2001-01-01", "endDate": "2002-12-31", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [10, 20]}
+            ],
+        ),
+        (
+            "4->2 condense (YYYY-MM dates)",
+            [
+                {"startDate": "2001-01", "endDate": "2001-12", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+                {"startDate": "2002-01", "endDate": "2002-03", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+            ],
+            [
+                {"startDate": "2001-01-01", "endDate": "2002-03-31", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]}
+            ],
+        ),
+        (
+            "2->1 condense (YYYY-MM-DD dates)",
+            [
+                {"startDate": "2001-01-01", "endDate": "2001-12-31", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+                {"startDate": "2002-01-01", "endDate": "2002-05-04", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]},
+            ],
+            [
+                {"startDate": "2001-01-01", "endDate": "2002-05-04", "term": {"@id": "treeNutTree", "units": "% area"},
+                 "value": [2]}
+            ],
+        ),
+        (
+            "3->1 condense",
+            [
+                {"startDate": "2001-01-01", "endDate": "2001-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]},
+                {"startDate": "2002-01-01", "endDate": "2002-10-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]},
+                {"startDate": "2002-11-01", "endDate": "2004-04-05", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]},
+            ],
+            [
+                {"startDate": "2001-01-01", "endDate": "2004-04-05", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]}
+            ],
+        ),
+        (
+            "3->2 partial condense",
+            [
+                {"startDate": "2001-01-01", "endDate": "2001-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]},
+                {"startDate": "2012-02-01", "endDate": "2012-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]},
+                {"startDate": "2002-01-01", "endDate": "2003-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]},
+            ],
+            [
+                {"startDate": "2001-01-01", "endDate": "2003-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]},
+                {"startDate": "2012-02-01", "endDate": "2012-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [9]}
+            ],
+        ),
+        (
+            "7->2 multi-condense",
+            [
+                {"startDate": "2001-01-01", "endDate": "2001-11-30", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [7]},
+                {"startDate": "2012-02-01", "endDate": "2012-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [7]},
+                {"startDate": "2001-12-01", "endDate": "2001-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [7]},
+                {"startDate": "2002-01-01", "endDate": "2002-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [7]},
+                {"startDate": "2013-01-01", "endDate": "2013-05-20", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [7]},
+            ],
+            [
+                {"startDate": "2001-01-01", "endDate": "2002-12-31", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [7]},
+                {"startDate": "2012-02-01", "endDate": "2013-05-20", "term": {"@id": "bananaPlant", "units": "% area"},
+                 "value": [7]}
+            ],
+        ),
+    ]
+)
+def test_condense_nodes(test_name, input_nodes, expected_output_nodes):
+    assert condense_nodes(input_nodes) == expected_output_nodes
+
+
 @patch(f"{CLASS_PATH}.download_hestia", side_effect=lambda id, *args: TERM_BY_ID[id])
 @patch(f"{CLASS_PATH}.related_cycles")
 def test_should_run(mock_related_cycles, *args):
@@ -121,7 +269,19 @@ def test_should_run(mock_related_cycles, *args):
         ("Example 1", f"{fixtures_folder}/inputs/example1"),
         ("Example 2", f"{fixtures_folder}/inputs/example2"),
         ("Example 3", f"{fixtures_folder}/inputs/example3"),
-        ("Example 4", f"{fixtures_folder}/inputs/example4")
+        ("Example 4", f"{fixtures_folder}/inputs/example4"),
+        ("Condense Nodes", f"{fixtures_folder}/inputs/condense_nodes")
+        # Expected:
+        #   - appleTree (81) x 3 condenses 2020-03-01 to 2021-02-15
+        #   - animalManureUsed (true) x 2 condenses 2001-04-01 to 2001-12-31
+        #   - treeNutTree, lebbekTree (82) does not condense [different terms]
+        #   - organicFertiliserUsed (true|false) does not condense [different values]
+        #   - glassOrHighAccessibleCover (83) does not condense [different date ranges (overlapping)]
+        #   - durianTree (84) does not condense [different date ranges (disjoint)]
+        #   - irrigatedSurfaceIrrigationContinuouslyFlooded (85) does not condense ["%" units]
+        #   - sassafrasTree (86) x 2 condenses 2001-01-01 to 2004-12-31
+        #   - bananaPlant (87) does not condense [non-consecutive years]
+        #   - durianTree (89) does not condense [dates overwritten See 808]
     ]
 )
 @patch(f"{CLASS_PATH}.download_hestia", side_effect=lambda id, *args: TERM_BY_ID[id])