hestia-earth-models 0.74.4__py3-none-any.whl → 0.74.5__py3-none-any.whl

tests/models/ipcc2019/test_co2ToAirBelowGroundBiomassStockChange.py

@@ -61,13 +61,11 @@ RUN_IDS = [f"{param[0]}, cycle{param[2]}" for param in RUN_PARAMS]
 @mark.parametrize("subfolder, num_cycles, cycle_index", RUN_PARAMS, ids=RUN_IDS)
 @patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
 @patch(f"{utils_path}.related_cycles")
-@patch(f"{utils_path}._get_site")
 @patch(f"{biomass_utils_path}.get_cover_crop_property_terms", return_value=COVER_CROP_PROPERTY_TERM_IDS)
 @patch(f"{term_path}.search")
 def test_run(
     search_mock: MagicMock,
     get_cover_crop_property_terms_mock: MagicMock,
-    get_site_mock: MagicMock,
     related_cycles_mock: MagicMock,
     _new_emission_mock: MagicMock,
     subfolder: str,
@@ -85,7 +83,7 @@ def test_run(
         _load_fixture(f"{fixtures_folder}/{subfolder}/cycle{i}.jsonld") for i in range(num_cycles)
     ]
 
-    get_site_mock.return_value = site
+    cycle["site"] = site
     related_cycles_mock.return_value = cycles
 
     result = run(cycle)
@@ -97,13 +95,11 @@ def test_run(
 
 @patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
 @patch(f"{utils_path}.related_cycles")
-@patch(f"{utils_path}._get_site")
 @patch(f"{biomass_utils_path}.get_cover_crop_property_terms", return_value=COVER_CROP_PROPERTY_TERM_IDS)
 @patch(f"{term_path}.search")
 def test_run_empty(
     search_mock: MagicMock,
     get_cover_crop_property_terms_mock: MagicMock,
-    get_site_mock: MagicMock,
     related_cycles_mock: MagicMock,
     _new_emission_mock: MagicMock
 ):
@@ -113,7 +109,6 @@ def test_run_empty(
     CYCLE = {}
     EXPECTED = []
 
-    get_site_mock.return_value = {}
     related_cycles_mock.return_value = [CYCLE]
 
     result = run(CYCLE)

tests/models/ipcc2019/test_co2ToAirBiocharStockChange.py

@@ -0,0 +1,90 @@
+from functools import reduce
+import json
+from os.path import isfile
+from pytest import mark
+from unittest.mock import MagicMock, patch
+
+from hestia_earth.models.ipcc2019.co2ToAirBiocharStockChange import MODEL, run
+
+from tests.utils import fake_new_emission, fixtures_path
+
+class_path = f"hestia_earth.models.{MODEL}.co2ToAirBiocharStockChange"
+utils_path = f"hestia_earth.models.{MODEL}.co2ToAirCarbonStockChange_utils"
+fixtures_folder = f"{fixtures_path}/{MODEL}/co2ToAirBiocharStockChange"
+
+
+def _load_fixture(path: str, default=None):
+    if isfile(path):
+        with open(path, encoding="utf-8") as f:
+            return json.load(f)
+    return default
+
+
+RUN_SCENARIOS = [
+    ("basic-case", 1),
+    ("no-overlapping-cycles", 3),
+    ("overlapping-cycles", 4),
+    ("complex-overlapping-cycles", 5),
+    ("missing-measurement-dates", 3),
+    ("no-biomass-measurements", 1),
+    ("non-consecutive-biomass-measurements", 1),
+    ("multiple-method-classifications", 5),
+    ("non-soil-based-gohac-system", 3),
+    ("with-gapfilled-start-date-end-date", 1)
+]
+"""List of (subfolder: str, num_cycles: int)."""
+
+RUN_PARAMS = reduce(
+    lambda params, scenario: params + [(scenario[0], scenario[1], i) for i in range(scenario[1])],
+    RUN_SCENARIOS,
+    list()
+)
+"""List of (subfolder: str, num_cycles: int, cycle_index: int)."""
+
+RUN_IDS = [f"{param[0]}, cycle{param[2]}" for param in RUN_PARAMS]
+
+
+@mark.parametrize("subfolder, num_cycles, cycle_index", RUN_PARAMS, ids=RUN_IDS)
+@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{utils_path}.related_cycles")
+def test_run(
+    related_cycles_mock: MagicMock,
+    _new_emission_mock: MagicMock,
+    subfolder: str,
+    num_cycles: int,
+    cycle_index: int
+):
+    """
+    Test `run` function for each cycle in each scenario.
+    """
+    site = _load_fixture(f"{fixtures_folder}/{subfolder}/site.jsonld")
+    cycle = _load_fixture(f"{fixtures_folder}/{subfolder}/cycle{cycle_index}.jsonld")
+    expected = _load_fixture(f"{fixtures_folder}/{subfolder}/result{cycle_index}.jsonld", default=[])
+
+    cycles = [
+        _load_fixture(f"{fixtures_folder}/{subfolder}/cycle{i}.jsonld") for i in range(num_cycles)
+    ]
+
+    cycle["site"] = site
+    related_cycles_mock.return_value = cycles
+
+    result = run(cycle)
+    assert result == expected
+
+
+@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{utils_path}.related_cycles")
+def test_run_empty(
+    related_cycles_mock: MagicMock,
+    _new_emission_mock: MagicMock
+):
+    """
+    Test `run` function for each cycle in each scenario.
+    """
+    CYCLE = {}
+    EXPECTED = []
+
+    related_cycles_mock.return_value = [CYCLE]
+
+    result = run(CYCLE)
+    assert result == EXPECTED

tests/models/ipcc2019/test_co2ToAirSoilOrganicCarbonStockChange.py

@@ -86,7 +86,6 @@ RUN_IDS = [f"{param[0]}, cycle{param[2]}" for param in RUN_PARAMS]
 @mark.parametrize("subfolder, num_cycles, cycle_index", RUN_PARAMS, ids=RUN_IDS)
 @patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
 @patch(f"{utils_path}.related_cycles")
-@patch(f"{utils_path}._get_site")
 @patch(f"{soc_tier_1_utils_path}.get_upland_rice_land_cover_terms", return_value=UPLAND_RICE_LAND_COVER_TERM_IDS)
 @patch(f"{soc_utils_path}.get_irrigated_terms", return_value=IRRIGATED_TERM_IDS)
 @patch(f"{soc_utils_path}.get_cover_crop_property_terms", return_value=COVER_CROP_PROPERTY_TERM_IDS)
@@ -96,7 +95,6 @@ def test_run(
     _get_cover_crop_property_terms_mock: MagicMock,
     _get_irrigated_terms_mock: MagicMock,
     _get_upland_rice_land_cover_terms_mock: MagicMock,
-    get_site_mock: MagicMock,
     related_cycles_mock: MagicMock,
     _new_emission_mock: MagicMock,
     subfolder: str,
@@ -114,7 +112,7 @@ def test_run(
         _load_fixture(f"{fixtures_folder}/{subfolder}/cycle{i}.jsonld") for i in range(num_cycles)
     ]
 
-    get_site_mock.return_value = site
+    cycle["site"] = site
     related_cycles_mock.return_value = cycles
 
     result = run(cycle)
@@ -126,7 +124,6 @@ def test_run(
 
 @patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
 @patch(f"{utils_path}.related_cycles")
-@patch(f"{utils_path}._get_site")
 @patch(f"{soc_tier_1_utils_path}.get_upland_rice_land_cover_terms", return_value=UPLAND_RICE_LAND_COVER_TERM_IDS)
 @patch(f"{soc_utils_path}.get_irrigated_terms", return_value=IRRIGATED_TERM_IDS)
 @patch(f"{soc_utils_path}.get_cover_crop_property_terms", return_value=COVER_CROP_PROPERTY_TERM_IDS)
@@ -136,7 +133,6 @@ def test_run_empty(
     _get_cover_crop_property_terms_mock: MagicMock,
     _get_irrigated_terms_mock: MagicMock,
     _get_upland_rice_land_cover_terms_mock: MagicMock,
-    get_site_mock: MagicMock,
     related_cycles_mock: MagicMock,
     _new_emission_mock: MagicMock
 ):
@@ -146,7 +142,6 @@ def test_run_empty(
     CYCLE = {}
     EXPECTED = []
 
-    get_site_mock.return_value = {}
     related_cycles_mock.return_value = [CYCLE]
 
     result = run(CYCLE)

tests/models/pooreNemecek2018/test_landOccupationDuringCycle.py

@@ -19,6 +19,7 @@ def test_should_run(mock_land_occupation, *args):
     assert not should_run
 
     # with product value and economicValueShare => run
+    impact['product'] = {'economicValueShare': 10}
     mock_land_occupation.return_value = 10
     should_run, *args = _should_run(impact)
     assert should_run is True

tests/orchestrator/strategies/merge/test_merge_list.py

@@ -1,6 +1,7 @@
 from unittest.mock import patch
 import pydash
 
+from hestia_earth.orchestrator.utils import reset_index
 from hestia_earth.orchestrator.strategies.merge.merge_list import merge, _get_value
 
 class_path = 'hestia_earth.orchestrator.strategies.merge.merge_list'
@@ -66,6 +67,7 @@ def test_merge_existing_node(*args):
         'depthUpper': 50
     }], version, model=model, node_type=node_type)
     assert len(result) == 3
+    reset_index()
 
     node_type = 'Cycle'
     model = {'key': 'emissions'}
@@ -79,6 +81,7 @@ def test_merge_existing_node(*args):
         'inputs': [{'@id': 'input-1'}]
     }], version, model=model, node_type=node_type)
     assert len(result) == 1
+    reset_index()
 
     # with different inputs => should not merge
     result = merge([{
@@ -89,6 +92,7 @@ def test_merge_existing_node(*args):
         'inputs': [{'@id': 'input-2'}]
     }], version, model=model, node_type=node_type)
     assert len(result) == 2
+    reset_index()
 
     result = merge([{
         **old_node,
@@ -98,6 +102,7 @@ def test_merge_existing_node(*args):
         'inputs': [{'@id': 'input-1'}, {'@id': 'input-2'}]
     }], version, model=model, node_type=node_type)
     assert len(result) == 2
+    reset_index()
 
     # with no inputs => should not merge
     result = merge([old_node], [{