hestia-earth-models 0.73.7__py3-none-any.whl → 0.74.0__py3-none-any.whl

tests/models/ipcc2019/test_n2OToAirOrganicFertiliserIndirect.py

@@ -2,40 +2,14 @@ from unittest.mock import patch
 import json
 from tests.utils import fixtures_path, fake_new_emission
 
-from hestia_earth.models.ipcc2019.n2OToAirOrganicFertiliserIndirect import (
-    MODEL, TERM_ID, run, _should_run, NH3_TERM_ID, NO3_TERM_ID, NOX_TERM_ID
-)
+from hestia_earth.models.ipcc2019 import MODEL
+from hestia_earth.models.ipcc2019.n2OToAirOrganicFertiliserIndirect import TERM_ID, run
 
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+utils_path = f"hestia_earth.models.{MODEL}.n2OToAir_indirect_emissions_utils"
 fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 
-def test_should_run():
-    # no emissions => no run
-    cycle = {'completeness': {'fertiliser': True}, 'emissions': []}
-    should_run, *args = _should_run(cycle)
-    assert not should_run
-
-    # with no3 emission => run
-    cycle['emissions'] = [
-        {
-            'term': {'@id': NO3_TERM_ID},
-            'value': [100]
-        },
-        {
-            'term': {'@id': NH3_TERM_ID},
-            'value': [100]
-        },
-        {
-            'term': {'@id': NOX_TERM_ID},
-            'value': [100]
-        }
-    ]
-    should_run, *args = _should_run(cycle)
-    assert should_run is True
-
-
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
 def test_run(*args):
     with open(f"{fixtures_folder}/cycle.jsonld", encoding='utf-8') as f:
         cycle = json.load(f)
@@ -47,7 +21,7 @@ def test_run(*args):
     assert value == expected
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
 def test_run_wet(*args):
     with open(f"{fixtures_folder}/ecoClimateZone-wet/cycle.jsonld", encoding='utf-8') as f:
         cycle = json.load(f)
@@ -59,7 +33,7 @@ def test_run_wet(*args):
     assert value == expected
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
 def test_run_dry(*args):
     with open(f"{fixtures_folder}/ecoClimateZone-dry/cycle.jsonld", encoding='utf-8') as f:
         cycle = json.load(f)

tests/models/ipcc2019/test_n2OToAirOrganicSoilBurningIndirect.py

@@ -0,0 +1,45 @@
+from unittest.mock import patch
+import json
+from tests.utils import fixtures_path, fake_new_emission
+
+from hestia_earth.models.ipcc2019 import MODEL
+from hestia_earth.models.ipcc2019.n2OToAirOrganicSoilBurningIndirect import TERM_ID, run
+
+utils_path = f"hestia_earth.models.{MODEL}.n2OToAir_indirect_emissions_utils"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
+def test_run(*args):
+    with open(f"{fixtures_folder}/cycle.jsonld", encoding='utf-8') as f:
+        cycle = json.load(f)
+
+    with open(f"{fixtures_folder}/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(cycle)
+    assert value == expected
+
+
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
+def test_run_wet(*args):
+    with open(f"{fixtures_folder}/ecoClimateZone-wet/cycle.jsonld", encoding='utf-8') as f:
+        cycle = json.load(f)
+
+    with open(f"{fixtures_folder}/ecoClimateZone-wet/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(cycle)
+    assert value == expected
+
+
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
+def test_run_dry(*args):
+    with open(f"{fixtures_folder}/ecoClimateZone-dry/cycle.jsonld", encoding='utf-8') as f:
+        cycle = json.load(f)
+
+    with open(f"{fixtures_folder}/ecoClimateZone-dry/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(cycle)
+    assert value == expected

tests/models/ipcc2019/test_n2OToAirOrganicSoilCultivationIndirect.py

@@ -0,0 +1,45 @@
+from unittest.mock import patch
+import json
+from tests.utils import fixtures_path, fake_new_emission
+
+from hestia_earth.models.ipcc2019 import MODEL
+from hestia_earth.models.ipcc2019.n2OToAirOrganicSoilCultivationIndirect import TERM_ID, run
+
+utils_path = f"hestia_earth.models.{MODEL}.n2OToAir_indirect_emissions_utils"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
+def test_run(*args):
+    with open(f"{fixtures_folder}/cycle.jsonld", encoding='utf-8') as f:
+        cycle = json.load(f)
+
+    with open(f"{fixtures_folder}/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(cycle)
+    assert value == expected
+
+
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
+def test_run_wet(*args):
+    with open(f"{fixtures_folder}/ecoClimateZone-wet/cycle.jsonld", encoding='utf-8') as f:
+        cycle = json.load(f)
+
+    with open(f"{fixtures_folder}/ecoClimateZone-wet/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(cycle)
+    assert value == expected
+
+
+@patch(f"{utils_path}._new_emission", side_effect=fake_new_emission)
+def test_run_dry(*args):
+    with open(f"{fixtures_folder}/ecoClimateZone-dry/cycle.jsonld", encoding='utf-8') as f:
+        cycle = json.load(f)
+
+    with open(f"{fixtures_folder}/ecoClimateZone-dry/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(cycle)
+    assert value == expected

tests/models/ipcc2019/test_n2OToAir_indirect_emissions_utils.py

@@ -0,0 +1,19 @@
+from hestia_earth.models.ipcc2019.n2OToAir_indirect_emissions_utils import _should_run
+from hestia_earth.models.ipcc2019.n2OToAirCropResidueDecompositionIndirect import _EMISSION_IDS
+
+
+def test_should_run():
+    # no emissions => no run
+    cycle = {'completeness': {'cropResidue': True}, 'emissions': []}
+    should_run, *args = _should_run('', _EMISSION_IDS, cycle)
+    assert not should_run
+
+    # with no3 emission => run
+    cycle['emissions'] = [
+        {
+            'term': {'@id': id},
+            'value': [100]
+        } for id in _EMISSION_IDS
+    ]
+    should_run, *args = _should_run('', _EMISSION_IDS, cycle)
+    assert should_run is True

tests/models/site/pre_checks/test_cache_geospatialDatabase.py

@@ -11,10 +11,10 @@ def test_list_rasters():
 @pytest.mark.parametrize(
     'name,site,expected_len',
     [
-        ('without values', {}, 3),
-        ('with region', {'region': 'region'}, 2),
-        ('with ecoregion', {'ecoregion': 'ecoregion'}, 2),
-        ('with awareWaterBasinId', {'awareWaterBasinId': 'awareWaterBasinId'}, 2),
+        ('without values', {}, 4),
+        ('with region', {'region': 'region'}, 3),
+        ('with ecoregion', {'ecoregion': 'ecoregion'}, 3),
+        ('with awareWaterBasinId', {'awareWaterBasinId': 'awareWaterBasinId'}, 3),
     ]
 )
 def test_list_vectors(name: str, site: dict, expected_len: int):

tests/models/test_config.py

@@ -4,8 +4,15 @@ import pytest
 from hestia_earth.utils.tools import flatten
 
 from hestia_earth.models.config import (
-    load_config, config_max_stage, _is_aggregated_model, _remove_aggregated, load_run_config, load_trigger_config,
-    get_max_stage
+    load_config,
+    config_max_stage,
+    _is_aggregated_model,
+    _remove_aggregated,
+    _use_aware_1,
+    load_run_config,
+    load_trigger_config,
+    get_max_stage,
+    AWARE_VERSION
 )
 
 _aggregated_model = {
@@ -29,6 +36,27 @@ def test_load_config_error():
         load_config(node_type)
 
 
+def test_load_config_skip_aggregated_models():
+    node_type = 'Cycle'
+    all_models = load_config(node_type, skip_aggregated_models=False).get('models')
+    models_no_aggregated = load_config(node_type, skip_aggregated_models=True).get('models')
+    assert all_models != models_no_aggregated
+
+
+def test_load_config_use_aware_v1_site():
+    node_type = 'Site'
+    v1_models = load_config(node_type, use_aware_version=AWARE_VERSION.V1).get('models')
+    v2_models = load_config(node_type, use_aware_version=AWARE_VERSION.V2).get('models')
+    assert v1_models != v2_models
+
+
+def test_load_config_use_aware_v1_impact_assessment():
+    node_type = 'ImpactAssessment'
+    v1_models = load_config(node_type, use_aware_version=AWARE_VERSION.V1).get('models')
+    v2_models = load_config(node_type, use_aware_version=AWARE_VERSION.V2).get('models')
+    assert v1_models != v2_models
+
+
 def test_config_max_stage():
     node_type = 'Cycle'
     config = load_config(node_type)
@@ -48,11 +76,29 @@ def test_remove_aggregated():
     assert _remove_aggregated(models) == [[_other_model], _other_model]
 
 
-def test_load_config_skip_aggregated_models():
-    node_type = 'Cycle'
-    all_models = load_config(node_type, skip_aggregated_models=False).get('models')
-    models_no_aggregated = load_config(node_type, skip_aggregated_models=True).get('models')
-    assert all_models != models_no_aggregated
+def test_use_aware_1():
+    scarcity_model = {
+        "model": "aware2-0",
+        "value": "scarcityWeightedWaterUse"
+    }
+    basinid_model = {
+        "model": "geospatialDatabase",
+        "value": "awareWaterBasinId"
+    }
+    models = [
+        [scarcity_model],
+        basinid_model
+    ]
+    assert _use_aware_1(models) == [
+        [{
+            "model": "aware",
+            "value": "scarcityWeightedWaterUse"
+        }],
+        {
+            "model": "geospatialDatabase",
+            "value": "awareWaterBasinId_v1"
+        }
+    ]
 
 
 def test_load_run_config():

tests/models/utils/test_background_emissions.py

@@ -0,0 +1,13 @@
+from hestia_earth.models.utils.background_emissions import _values_from_column
+
+
+def test_values_from_column():
+    column = 'landTransformation+landCover[pond]+previousLandCover[permanentCropland]+country[GADM-BRA]'
+    assert _values_from_column('', column, '10') == {
+        'landTransformation': {
+            'value': 10,
+            'landCover': 'pond',
+            'previousLandCover': 'permanentCropland',
+            'country': 'GADM-BRA'
+        }
+    }