hestia-earth-models 0.64.12__py3-none-any.whl → 0.64.14__py3-none-any.whl

tests/models/utils/test_site.py

@@ -1,7 +1,8 @@
-from unittest.mock import patch
+import pytest
+from unittest.mock import Mock, patch
 from hestia_earth.schema import SiteSiteType
 
-from hestia_earth.models.utils.site import region_level_1_id, related_cycles, valid_site_type
+from hestia_earth.models.utils.site import region_level_1_id, related_cycles, valid_site_type, get_land_cover_term_id
 
 class_path = 'hestia_earth.models.utils.site'
 CYCLE = {'@id': 'id'}
@@ -30,3 +31,33 @@ def test_valid_site_type():
 
     site = {'siteType': SiteSiteType.CROPLAND.value}
     assert not valid_site_type(site, [SiteSiteType.OTHER_NATURAL_VEGETATION.value])
+
+
+_LAND_COVER_TERMS = [
+    {'@type': 'Term', 'name': 'Glass or high accessible cover', '@id': 'glassOrHighAccessibleCover'},
+    {'@type': 'Term', 'name': 'Sea or ocean', '@id': 'seaOrOcean'},
+    {'@type': 'Term', 'name': 'River or stream', '@id': 'riverOrStream'},
+    {'@type': 'Term', 'name': 'Other natural vegetation', '@id': 'otherNaturalVegetation'},
+    {'@type': 'Term', 'name': 'Food retailer', '@id': 'foodRetailer'},
+    {'@type': 'Term', 'name': 'Agri-food processor', '@id': 'agriFoodProcessor'},
+    {'@type': 'Term', 'name': 'Permanent pasture', '@id': 'permanentPasture'},
+    {'@type': 'Term', 'name': 'Animal housing', '@id': 'animalHousing'},
+    {'@type': 'Term', 'name': 'Forest', '@id': 'forest'},
+    {'@type': 'Term', 'name': 'Lake', '@id': 'lake'},
+    {'@type': 'Term', 'name': 'Pond', '@id': 'pond'},
+    {'@type': 'Term', 'name': 'Cropland', '@id': 'cropland'},
+]
+
+
+@pytest.mark.parametrize(
+    'site_type,expected_landCover_id',
+    [
+        (SiteSiteType.CROPLAND.value, 'cropland'),
+        (SiteSiteType.SEA_OR_OCEAN.value, 'seaOrOcean'),
+        (SiteSiteType.AGRI_FOOD_PROCESSOR.value, 'agriFoodProcessor'),
+        (SiteSiteType.GLASS_OR_HIGH_ACCESSIBLE_COVER.value, 'glassOrHighAccessibleCover'),
+    ]
+)
+@patch(f"{class_path}.get_land_cover_siteTypes", return_value=_LAND_COVER_TERMS)
+def test_get_land_cover_term_id(mock_terms: Mock, site_type: str, expected_landCover_id: str):
+    assert get_land_cover_term_id(site_type) == expected_landCover_id, site_type

tests/models/pooreNemecek2018/test_no3ToGroundwaterSoilFlux.py

@@ -1,90 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fixtures_path, fake_new_emission, FLOODED_RICE_TERMS
-
-from hestia_earth.models.pooreNemecek2018.no3ToGroundwaterSoilFlux import MODEL, TERM_ID, run, _should_run
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-@patch(f"{class_path}.get_rice_paddy_terms", return_value=FLOODED_RICE_TERMS)
-@patch(f"{class_path}.find_primary_product", return_value={'term': {'@id': 'product'}})
-@patch(f"{class_path}.get_crop_residue_decomposition_N_total", return_value=10)
-@patch(f"{class_path}.most_relevant_measurement_value", return_value=0)
-@patch(f"{class_path}.get_max_rooting_depth", return_value=None)
-def test_should_run(mock_rooting_depth, mock_measurement, *args):
-    # no measurements => no run
-    cycle = {}
-    should_run, *args = _should_run(cycle)
-    assert not should_run
-
-    # with measurements => no run
-    mock_measurement.return_value = 10
-    should_run, *args = _should_run(cycle)
-    assert not should_run
-
-    # no rootingDepth => no run
-    should_run, *args = _should_run({})
-    assert not should_run
-
-    # with rootingDepth => run
-    mock_rooting_depth.return_value = 10
-    should_run, *args = _should_run({})
-    assert should_run is True
-
-
-@patch(f"{class_path}.get_rice_paddy_terms", return_value=FLOODED_RICE_TERMS)
-@patch(f"{class_path}.get_max_rooting_depth", return_value=0.9)
-@patch(f"{class_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"{class_path}.get_rice_paddy_terms", return_value=FLOODED_RICE_TERMS)
-@patch(f"{class_path}.get_max_rooting_depth", return_value=0.9)
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
-def test_run_with_high_conditions(*args):
-    with open(f"{fixtures_folder}/with-high-conditions/cycle.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
-
-    with open(f"{fixtures_folder}/with-high-conditions/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    value = run(cycle)
-    assert value == expected
-
-
-@patch(f"{class_path}.get_rice_paddy_terms", return_value=FLOODED_RICE_TERMS)
-@patch(f"{class_path}.get_max_rooting_depth", return_value=0.9)
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
-def test_run_with_low_conditions(*args):
-    with open(f"{fixtures_folder}/with-low-conditions/cycle.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
-
-    with open(f"{fixtures_folder}/with-low-conditions/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    value = run(cycle)
-    assert value == expected
-
-
-@patch(f"{class_path}.get_rice_paddy_terms", return_value=FLOODED_RICE_TERMS)
-@patch(f"{class_path}.get_max_rooting_depth", return_value=0.9)
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
-def test_run_with_flooded_rice(*args):
-    with open(f"{fixtures_folder}/with-flooded-rice/cycle.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
-
-    with open(f"{fixtures_folder}/with-flooded-rice/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    value = run(cycle)
-    assert value == expected

tests/models/stehfestBouwman2006/test_n2OToAirSoilFlux.py

@@ -1,41 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fixtures_path, fake_new_emission
-
-from hestia_earth.models.stehfestBouwman2006.n2OToAirSoilFlux import MODEL, TERM_ID, run, _should_run, _get_value
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-@patch(f"{class_path}.find_primary_product", return_value={'@id': 'product'})
-@patch(f"{class_path}._get_crop_crouping", return_value='Cereals')
-@patch(f"{class_path}.get_crop_residue_decomposition_N_total", return_value=10)
-@patch(f"{class_path}.most_relevant_measurement_value", return_value=0)
-def test_should_run(mock_measurement, *args):
-    # no measurements => no run
-    cycle = {}
-    should_run, *args = _should_run(cycle)
-    assert not should_run
-
-    # with measurements => run
-    mock_measurement.return_value = 10
-    should_run, *args = _should_run(cycle)
-    assert should_run is True
-
-
-def test_handle_overflow_error():
-    value = _get_value({}, [10, 10, 10, 10, 1, 'Other'], 10000000)
-    assert round(value) == 1131197
-
-
-@patch(f"{class_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

tests/models/stehfestBouwman2006/test_noxToAirSoilFlux.py

@@ -1,40 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fixtures_path, fake_new_emission
-
-from hestia_earth.models.stehfestBouwman2006.noxToAirSoilFlux import MODEL, TERM_ID, run, _should_run, _get_value
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-@patch(f"{class_path}.get_crop_residue_decomposition_N_total", return_value=10)
-@patch(f"{class_path}.most_relevant_measurement_value", return_value=0)
-def test_should_run(mock_measurement, *args):
-    # no measurements => no run
-    cycle = {'inputs': [], 'measurements': []}
-    should_run, *args = _should_run(cycle)
-    assert not should_run
-
-    # with measurements => run
-    mock_measurement.return_value = 10
-    cycle['measurements'] = [{}]
-    should_run, *args = _should_run(cycle)
-    assert should_run is True
-
-
-def test_handle_overflow_error():
-    value = _get_value({}, 1, 100, 10000000)
-    assert round(value) == 535554
-
-
-@patch(f"{class_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

tests/models/stehfestBouwman2006GisImplementation/test_noxToAirSoilFlux.py

@@ -1,33 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fixtures_path, fake_new_emission
-
-from hestia_earth.models.stehfestBouwman2006GisImplementation.noxToAirSoilFlux import MODEL, TERM_ID, run, _should_run
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-@patch(f"{class_path}.get_crop_residue_decomposition_N_total", return_value=10)
-def test_should_run(*args):
-    # no country => no run
-    cycle = {'inputs': [], 'site': {}}
-    should_run, *args = _should_run(cycle)
-    assert not should_run
-
-    # with country => run
-    cycle['site'] = {'country': {'@id': 'country'}}
-    should_run, *args = _should_run(cycle)
-    assert should_run is True
-
-
-@patch(f"{class_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