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

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/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