hestia-earth-models 0.64.7__py3-none-any.whl → 0.64.9__py3-none-any.whl

tests/models/ipcc2019/animal/test_trueProteinContent.py

@@ -0,0 +1,22 @@
+from unittest.mock import patch
+import json
+from tests.utils import fixtures_path, fake_new_property
+
+from hestia_earth.models.ipcc2019 import MODEL
+from hestia_earth.models.ipcc2019.animal.trueProteinContent import TERM_ID, run
+
+utils_path = '.'.join(['hestia_earth', 'models', MODEL, 'animal', 'utils'])
+class_path = f"hestia_earth.models.{MODEL}.animal.{TERM_ID}"
+fixtures_folder = f"{fixtures_path}/{MODEL}/animal/{TERM_ID}"
+
+
+@patch(f"{utils_path}._new_property", side_effect=fake_new_property)
+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/ipcc2019/animal/test_weightAtMaturity.py

@@ -2,7 +2,8 @@ from unittest.mock import patch
 import json
 from tests.utils import fixtures_path, fake_new_property
 
-from hestia_earth.models.ipcc2019.animal.weightAtMaturity import MODEL, TERM_ID, run
+from hestia_earth.models.ipcc2019 import MODEL
+from hestia_earth.models.ipcc2019.animal.weightAtMaturity import TERM_ID, run
 
 class_path = f"hestia_earth.models.{MODEL}.animal.{TERM_ID}"
 fixtures_folder = f"{fixtures_path}/{MODEL}/animal/{TERM_ID}"

tests/models/ipcc2019/test_co2ToAirAboveGroundBiomassStockChangeLandUseChange.py

@@ -0,0 +1,83 @@
+from functools import reduce
+import json
+from os.path import isfile
+from pytest import mark
+from unittest.mock import patch
+
+from hestia_earth.models.ipcc2019.co2ToAirAboveGroundBiomassStockChangeLandUseChange import MODEL, run, TERM_ID
+
+from tests.utils import fake_new_emission, fixtures_path
+
+class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+utils_path = f"hestia_earth.models.{MODEL}.co2ToAirCarbonStockChange_utils"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+RUN_SCENARIOS = [
+    ("no-overlapping-cycles", 3),
+    ("overlapping-cycles", 4),
+    ("complex-overlapping-cycles", 5),
+    ("missing-measurement-dates", 3),
+    ("no-biomass-measurements", 1),               # Closes issue #700
+    ("non-consecutive-biomass-measurements", 1),  # Closes issue #827
+    ("multiple-method-classifications", 5),              # Closes issue #764
+    ("non-soil-based-gohac-system", 3),                  # Closes issue #848
+    ("soil-based-gohac-system", 3)                       # Closes issue #848
+]
+"""List of (subfolder: str, num_cycles: int)."""
+
+
+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_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")
+@patch(f"{utils_path}._get_site")  # TODO: rationalise order of patches
+def test_run(_get_site_mock, related_cycles_mock, _new_emission_mock, subfolder, num_cycles, cycle_index):
+    """
+    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)
+    ]
+
+    _get_site_mock.return_value = 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")
+@patch(f"{utils_path}._get_site")  # TODO: rationalise order of patches
+def test_run_empty(_get_site_mock, related_cycles_mock, _new_emission_mock):
+    """
+    Test `run` function for each cycle in each scenario.
+    """
+    CYCLE = {}
+    EXPECTED = []
+
+    _get_site_mock.return_value = {}
+    related_cycles_mock.return_value = [CYCLE]
+
+    result = run(CYCLE)
+    assert result == EXPECTED

tests/models/ipcc2019/test_co2ToAirBelowGroundBiomassStockChangeLandUseChange.py

@@ -0,0 +1,83 @@
+from functools import reduce
+import json
+from os.path import isfile
+from pytest import mark
+from unittest.mock import patch
+
+from hestia_earth.models.ipcc2019.co2ToAirBelowGroundBiomassStockChangeLandUseChange import MODEL, run, TERM_ID
+
+from tests.utils import fake_new_emission, fixtures_path
+
+class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+utils_path = f"hestia_earth.models.{MODEL}.co2ToAirCarbonStockChange_utils"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+RUN_SCENARIOS = [
+    ("no-overlapping-cycles", 3),
+    ("overlapping-cycles", 4),
+    ("complex-overlapping-cycles", 5),
+    ("missing-measurement-dates", 3),
+    ("no-biomass-measurements", 1),               # Closes issue #700
+    ("non-consecutive-biomass-measurements", 1),  # Closes issue #827
+    ("multiple-method-classifications", 5),              # Closes issue #764
+    ("non-soil-based-gohac-system", 3),                  # Closes issue #848
+    ("soil-based-gohac-system", 3)                       # Closes issue #848
+]
+"""List of (subfolder: str, num_cycles: int)."""
+
+
+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_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")
+@patch(f"{utils_path}._get_site")
+def test_run(_get_site_mock, related_cycles_mock, _new_emission_mock, subfolder, num_cycles, cycle_index):
+    """
+    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)
+    ]
+
+    _get_site_mock.return_value = 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")
+@patch(f"{utils_path}._get_site")
+def test_run_empty(_get_site_mock, related_cycles_mock, _new_emission_mock):
+    """
+    Test `run` function for each cycle in each scenario.
+    """
+    CYCLE = {}
+    EXPECTED = []
+
+    _get_site_mock.return_value = {}
+    related_cycles_mock.return_value = [CYCLE]
+
+    result = run(CYCLE)
+    assert result == EXPECTED

tests/models/ipcc2019/test_co2ToAirCarbonStockChange_utils.py

@@ -1,8 +1,8 @@
 from hestia_earth.schema import EmissionMethodTier, MeasurementMethodClassification
 
 from hestia_earth.models.ipcc2019.co2ToAirCarbonStockChange_utils import (
-    add_carbon_stock_change_emissions, calc_carbon_stock_change, calc_carbon_stock_change_emission, _convert_c_to_co2,
-    lerp_carbon_stocks, CarbonStock, CarbonStockChange, CarbonStockChangeEmission
+    _add_carbon_stock_change_emissions, _calc_carbon_stock_change, _calc_carbon_stock_change_emission,
+    _convert_c_to_co2, _lerp_carbon_stocks, CarbonStock, CarbonStockChange, CarbonStockChangeEmission
 )
 
 
@@ -20,7 +20,7 @@ def test_lerp_carbon_stocks():
         21000, "2001-12-31", MeasurementMethodClassification.ON_SITE_PHYSICAL_MEASUREMENT
     )
 
-    result = lerp_carbon_stocks(START, END, TARGET_DATE)
+    result = _lerp_carbon_stocks(START, END, TARGET_DATE)
     assert result == EXPECTED
 
 
@@ -29,7 +29,7 @@ def test_calc_carbon_stock_change():
     END = CarbonStock(21000,  "2001", MeasurementMethodClassification.TIER_1_MODEL)
     EXPECTED = CarbonStockChange(1000, "2000", "2001", MeasurementMethodClassification.TIER_1_MODEL)
 
-    result = calc_carbon_stock_change(START, END)
+    result = _calc_carbon_stock_change(START, END)
     assert result == EXPECTED
 
 
@@ -37,7 +37,7 @@ def test_calc_carbon_stock_change_emission():
     SOC_STOCK_CHANGE = CarbonStockChange(-1000, "2000", "2001", MeasurementMethodClassification.TIER_1_MODEL)
     EXPECTED = CarbonStockChangeEmission(3663.836163836164, "2000", "2001", EmissionMethodTier.TIER_1)
 
-    result = calc_carbon_stock_change_emission(SOC_STOCK_CHANGE)
+    result = _calc_carbon_stock_change_emission(SOC_STOCK_CHANGE)
     assert result == EXPECTED
 
 
@@ -46,5 +46,5 @@ def test_add_carbon_stock_change_emissions():
     EMISSION_2 = CarbonStockChangeEmission(2000, "2001", "2002", EmissionMethodTier.TIER_1)
     EXPECTED = CarbonStockChangeEmission(5000, "2000", "2002", EmissionMethodTier.TIER_1)
 
-    result = add_carbon_stock_change_emissions(EMISSION_1, EMISSION_2)
+    result = _add_carbon_stock_change_emissions(EMISSION_1, EMISSION_2)
     assert result == EXPECTED

tests/models/ipcc2019/test_co2ToAirSoilOrganicCarbonStockChangeManagementChange.py

@@ -9,6 +9,7 @@ from hestia_earth.models.ipcc2019.co2ToAirSoilOrganicCarbonStockChangeManagement
 from tests.utils import fake_new_emission, fixtures_path
 
 class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+utils_path = f"hestia_earth.models.{MODEL}.co2ToAirCarbonStockChange_utils"
 fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 RUN_SCENARIOS = [
@@ -44,8 +45,8 @@ 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"{class_path}.related_cycles")
-@patch(f"{class_path}._get_site")
+@patch(f"{utils_path}.related_cycles")
+@patch(f"{utils_path}._get_site")
 def test_run(_get_site_mock, related_cycles_mock, _new_emission_mock, subfolder, num_cycles, cycle_index):
     """
     Test `run` function for each cycle in each scenario.
@@ -66,8 +67,8 @@ def test_run(_get_site_mock, related_cycles_mock, _new_emission_mock, subfolder,
 
 
 @patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
-@patch(f"{class_path}.related_cycles")
-@patch(f"{class_path}._get_site")
+@patch(f"{utils_path}.related_cycles")
+@patch(f"{utils_path}._get_site")
 def test_run_empty(_get_site_mock, related_cycles_mock, _new_emission_mock):
     """
     Test `run` function for each cycle in each scenario.

tests/models/poschEtAl2008/test_terrestrialAcidificationPotentialAccumulatedExceedance.py

@@ -1,45 +1,58 @@
 import json
 from unittest.mock import patch
 
-from hestia_earth.models.poschEtAl2008.terrestrialAcidificationPotentialAccumulatedExceedance import MODEL, TERM_ID, run
+from hestia_earth.models.poschEtAl2008.terrestrialAcidificationPotentialAccumulatedExceedance import MODEL, TERM_ID, \
+    run
 from tests.utils import fixtures_path, fake_new_indicator
 
 class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
 fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 
+def fake_rounded_indicator(value: float):
+    indicator = fake_new_indicator(TERM_ID, MODEL)
+    indicator['value'] = round(value, 7)
+    return indicator
+
+
 @patch(f"{class_path}._new_indicator", side_effect=fake_new_indicator)
 def test_run(*args):
     with open(f"{fixtures_folder}/impact-assessment.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
+        impactassessment = json.load(f)
 
-    with open(f"{fixtures_folder}/result.jsonld", encoding='utf-8') as f:
+    with open(f"{fixtures_folder}/Australia/result.jsonld", encoding='utf-8') as f:
         expected = json.load(f)
 
-    value = run(cycle)
+    value = run(impactassessment)
     assert value == expected
 
 
-@patch(f"{class_path}._new_indicator", side_effect=fake_new_indicator)
+@patch(f"{class_path}._indicator", side_effect=fake_rounded_indicator)
 def test_lookup_to_bad_country(*args):
+    """
+    Should default to region-world value
+    """
     with open(f"{fixtures_folder}/impact-assessment.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
-    cycle['country']['@id'] = "example-land-not-real"
+        impactassessment = json.load(f)
 
-    value = run(cycle)
-    assert value['value'] is None
+    impactassessment['country']['@id'] = "example-land-not-real"
+
+    with open(f"{fixtures_folder}/region-world/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(impactassessment)
+    assert value == expected
 
 
 @patch(f"{class_path}._new_indicator", side_effect=fake_new_indicator)
-def test_lookup_no_term_type(*args):
+def test_run_no_emissions(*args):
     """
-    We currently do not filter out bad termTypes so this is expected behavior.
+    Impact assessment with no emissions should return a indicator of 0
     """
     with open(f"{fixtures_folder}/impact-assessment.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
-    del cycle['emissionsResourceUse'][0]['term']['termType']
-    del cycle['emissionsResourceUse'][1]['term']['termType']
-    cycle['emissionsResourceUse'][2]['term']['termType'] = "wrong-type"
+        impactassessment = json.load(f)
+
+    del impactassessment['emissionsResourceUse']
 
-    value = run(cycle)
-    assert value['value'] == 0.085
+    value = run(impactassessment)
+    assert value['value'] == 0

tests/models/poschEtAl2008/test_terrestrialEutrophicationPotentialAccumulatedExceedance.py

@@ -9,36 +9,50 @@ class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
 fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 
+def fake_rounded_indicator(value: float):
+    indicator = fake_new_indicator(TERM_ID, MODEL)
+    indicator['value'] = round(value, 7)
+    return indicator
+
+
 @patch(f"{class_path}._new_indicator", side_effect=fake_new_indicator)
 def test_run(*args):
     with open(f"{fixtures_folder}/impact-assessment.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
+        impactassessment = json.load(f)
 
-    with open(f"{fixtures_folder}/result.jsonld", encoding='utf-8') as f:
+    with open(f"{fixtures_folder}/Belarus/result.jsonld", encoding='utf-8') as f:
         expected = json.load(f)
 
-    value = run(cycle)
+    value = run(impactassessment)
     assert value == expected
 
 
-@patch(f"{class_path}._new_indicator", side_effect=fake_new_indicator)
+@patch(f"{class_path}._indicator", side_effect=fake_rounded_indicator)
 def test_lookup_to_bad_country(*args):
+    """
+    Should default to region-world value
+    """
     with open(f"{fixtures_folder}/impact-assessment.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
-    cycle['country']['@id'] = "example-land-not-real"
+        impactassessment = json.load(f)
 
-    value = run(cycle)
-    assert value['value'] is None
+    impactassessment['country']['@id'] = "example-land-not-real"
+
+    with open(f"{fixtures_folder}/region-world/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(impactassessment)
+    assert value == expected
 
 
 @patch(f"{class_path}._new_indicator", side_effect=fake_new_indicator)
-def test_lookup_no_term_type(*args):
+def test_run_no_emissions(*args):
     """
-    We currently do not filter out bad termTypes so this is expected behavior.
+    Impact assessment with no emissions should return a indicator of 0
     """
     with open(f"{fixtures_folder}/impact-assessment.jsonld", encoding='utf-8') as f:
-        cycle = json.load(f)
-    del cycle['emissionsResourceUse'][0]['term']['termType']
+        impactassessment = json.load(f)
+
+    del impactassessment['emissionsResourceUse']
 
-    value = run(cycle)
-    assert value['value'] == 7.47
+    value = run(impactassessment)
+    assert value['value'] == 0

tests/models/site/test_management.py

@@ -25,7 +25,10 @@ TERM_BY_ID = {
     }
 }
 
-LAND_COVER_TERM_BY_SITE_TYPE = {SiteSiteType.ANIMAL_HOUSING.value: "animalHousing"}
+LAND_COVER_TERM_BY_SITE_TYPE = {
+    SiteSiteType.ANIMAL_HOUSING.value: "animalHousing",
+    SiteSiteType.CROPLAND.value: "cropland"
+}
 
 
 def lookup_side_effect(*args, **kwargs):

tests/models/utils/test_input.py

@@ -1,7 +1,9 @@
+import pytest
 from unittest.mock import patch
+from hestia_earth.schema import TermTermType
 from tests.utils import TERM
 
-from hestia_earth.models.utils.input import _new_input
+from hestia_earth.models.utils.input import _new_input, get_feed_inputs
 
 class_path = 'hestia_earth.models.utils.input'
 
@@ -22,3 +24,65 @@ def test_new_input(*args):
         '@type': 'Input',
         'term': TERM
     }
+
+
+@pytest.mark.parametrize(
+    'test_name,cycle,expected_input_length',
+    [
+        (
+            'no inputs',
+            {},
+            0
+        ),
+        (
+            'with crop feed',
+            {
+                'inputs': [{
+                    'term': {'units': 'kg', 'termType': TermTermType.CROP.value, '@id': 'wheatGrain'},
+                    'isAnimalFeed': True,
+                    'value': [10]
+                }]
+            },
+            1
+        ),
+        (
+            'with crop no feed',
+            {
+                'inputs': [{
+                    'term': {'units': 'kg', 'termType': TermTermType.CROP.value, '@id': 'wheatGrain'},
+                    'isAnimalFeed': False,
+                    'value': [10]
+                }]
+            },
+            0
+        ),
+        (
+            'with feed food additive and energy content',
+            {
+                'inputs': [{
+                    'term': {
+                        'units': 'kg', 'termType': TermTermType.FEEDFOODADDITIVE.value, '@id': 'aminoAcidsUnspecified'
+                    },
+                    'isAnimalFeed': True,
+                    'value': [10]
+                }]
+            },
+            1
+        ),
+        (
+            'with feed food additive not energy content',
+            {
+                'inputs': [{
+                    'term': {
+                        'units': 'kg', 'termType': TermTermType.FEEDFOODADDITIVE.value, '@id': 'premixUnspecified'
+                    },
+                    'isAnimalFeed': True,
+                    'value': [10]
+                }]
+            },
+            0
+        ),
+    ]
+)
+def test_get_feed_inputs(test_name, cycle, expected_input_length):
+    assert len(get_feed_inputs(cycle)) == expected_input_length, test_name