hestia-earth-models 0.62.4__py3-none-any.whl → 0.62.6__py3-none-any.whl

tests/models/emepEea2019/test_nh3ToAirInorganicFertiliser.py

@@ -5,6 +5,7 @@ from tests.utils import fixtures_path, fake_new_emission
 from hestia_earth.models.emepEea2019.nh3ToAirInorganicFertiliser import MODEL, TERM_ID, run, _should_run
 
 class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+model_utils_path = f"hestia_earth.models.{MODEL}.utils"
 fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 
@@ -27,7 +28,7 @@ def test_should_run(mock_measurement, *args):
     assert should_run is True
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{model_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)
@@ -39,7 +40,7 @@ def test_run(*args):
     assert value == expected
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{model_utils_path}._new_emission", side_effect=fake_new_emission)
 def test_run_with_unspecified(*args):
     with open(f"{fixtures_folder}/with-unspecified/cycle.jsonld", encoding='utf-8') as f:
         cycle = json.load(f)
@@ -51,7 +52,7 @@ def test_run_with_unspecified(*args):
     assert value == expected
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{model_utils_path}._new_emission", side_effect=fake_new_emission)
 def test_run_complete(*args):
     with open(f"{fixtures_folder}/is-complete/cycle.jsonld", encoding='utf-8') as f:
         cycle = json.load(f)

tests/models/emepEea2019/test_noxToAirFuelCombustion.py

@@ -5,6 +5,7 @@ from tests.utils import fixtures_path, fake_new_emission
 from hestia_earth.models.emepEea2019.noxToAirFuelCombustion import MODEL, TERM_ID, run, _should_run
 
 class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+model_utils_path = f"hestia_earth.models.{MODEL}.utils"
 fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 
@@ -21,7 +22,7 @@ def test_should_run(mock_get_fuel_values):
     assert should_run is True
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{model_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)
@@ -33,7 +34,7 @@ def test_run(*args):
     assert value == expected
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{model_utils_path}._new_emission", side_effect=fake_new_emission)
 def test_run_data_complete(*args):
     with open(f"{fixtures_folder}/no-input-data-complete/cycle.jsonld", encoding='utf-8') as f:
         cycle = json.load(f)

tests/models/emepEea2019/test_pm10ToAirAnimalHousing.py

@@ -0,0 +1,21 @@
+from unittest.mock import patch
+import json
+
+from tests.utils import fixtures_path, fake_new_emission
+from hestia_earth.models.emepEea2019.pm10ToAirAnimalHousing import MODEL, TERM_ID, run
+
+class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+model_utils_path = f"hestia_earth.models.{MODEL}.utils"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+
+@patch(f"{model_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

tests/models/emepEea2019/test_pm25ToAirAnimalHousing.py

@@ -0,0 +1,21 @@
+from unittest.mock import patch
+import json
+
+from tests.utils import fixtures_path, fake_new_emission
+from hestia_earth.models.emepEea2019.pm25ToAirAnimalHousing import MODEL, TERM_ID, run
+
+class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+model_utils_path = f"hestia_earth.models.{MODEL}.utils"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+
+@patch(f"{model_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

tests/models/emepEea2019/test_so2ToAirFuelCombustion.py

@@ -5,6 +5,7 @@ from tests.utils import fixtures_path, fake_new_emission
 from hestia_earth.models.emepEea2019.so2ToAirFuelCombustion import MODEL, TERM_ID, run, _should_run
 
 class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+model_utils_path = f"hestia_earth.models.{MODEL}.utils"
 fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 
@@ -21,7 +22,7 @@ def test_should_run(mock_get_fuel_values):
     assert should_run is True
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{model_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)
@@ -33,7 +34,7 @@ def test_run(*args):
     assert value == expected
 
 
-@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+@patch(f"{model_utils_path}._new_emission", side_effect=fake_new_emission)
 def test_run_data_complete(*args):
     with open(f"{fixtures_folder}/no-input-data-complete/cycle.jsonld", encoding='utf-8') as f:
         cycle = json.load(f)

tests/models/emepEea2019/test_tspToAirAnimalHousing.py

@@ -0,0 +1,21 @@
+from unittest.mock import patch
+import json
+
+from tests.utils import fixtures_path, fake_new_emission
+from hestia_earth.models.emepEea2019.tspToAirAnimalHousing import MODEL, TERM_ID, run
+
+class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+model_utils_path = f"hestia_earth.models.{MODEL}.utils"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+
+@patch(f"{model_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

tests/models/emepEea2019/test_utils.py

@@ -1,6 +1,8 @@
 from unittest.mock import patch
 
-from hestia_earth.models.emepEea2019.utils import get_fuel_values
+import pytest
+
+from hestia_earth.models.emepEea2019.utils import get_fuel_values, should_run_animal, _duration_in_housing
 
 class_path = 'hestia_earth.models.emepEea2019.utils'
 TERMS = [
@@ -53,3 +55,198 @@ def test_get_fuel_values(*args):
     }
     result = get_fuel_values('co2ToAirFuelCombustion', cycle, 'co2ToAirFuelCombustionEmepEea2019')
     assert result == [317.0, 632.0, 475.5, 158.5]
+
+
+@pytest.mark.parametrize(
+    "test_name,cycle,expected_duration",
+    [
+        (
+            "no sites => 0",
+            {"completeness": {"@type": "Completeness", "animalPopulation": False}},
+            0
+        ),
+        (
+            "site and other sites have wrong type => 0",
+            {
+                "site": {"@type": "Site", "siteType": "permanent pasture"},
+                "siteDuration": 15,
+                "otherSites": [
+                    {"@type": "Site", "siteType": "pond"},
+                    {"@type": "Site", "siteType": "pond"},
+                    {"@type": "Site", "siteType": "pond"},
+                ],
+                "otherSitesDuration": [20, 21, 22],
+            },
+            0
+        ),
+        (
+            "only second other site is animal housing => 12",
+            {
+                "site": {"@type": "Site", "siteType": "pond"},
+                "siteDuration": 20,
+                "otherSites": [
+                    {"@type": "Site", "siteType": "pond"},
+                    {"@type": "Site", "siteType": "animal housing"},
+                    {"@type": "Site", "siteType": "pond"},
+                ],
+                "otherSitesDuration": [11, 12, 13],
+            },
+            12
+        ),
+        (
+            "only site is animal housing => 20",
+            {
+                "site": {"@type": "Site", "siteType": "animal housing"},
+                "otherSites": [
+                    {"@type": "Site", "siteType": "forest"},
+                    {"@type": "Site", "siteType": "forest"},
+                    {"@type": "Site", "siteType": "forest"},
+                ],
+                "siteDuration": 20,
+                "otherSitesDuration": [12, 13, 14],
+            },
+            20
+        ),
+        (
+            "site and otherSites are animal housing, sum all values => 59",
+            {
+                "site": {"@type": "Site", "siteType": "animal housing"},
+                "otherSites": [
+                    {"@type": "Site", "siteType": "animal housing"},
+                    {"@type": "Site", "siteType": "animal housing"},
+                    {"@type": "Site", "siteType": "animal housing"},
+                ],
+                "siteDuration": 20,
+                "otherSitesDuration": [12, 13, 14],
+            },
+            59
+        )
+
+    ]
+)
+def test_duration_in_housing(test_name, cycle, expected_duration):
+    assert _duration_in_housing(cycle) == expected_duration
+
+
+@pytest.mark.parametrize(
+    "test_name,cycle,expected_should_run",
+    [
+        (
+            "not complete => no run",
+            {"completeness": {"@type": "Completeness", "animalPopulation": False}},
+            False
+        ),
+        (
+            "no site or other sites => no run",
+            {"completeness": {"@type": "Completeness", "animalPopulation": True}},
+            False
+        ),
+        (
+            "no site or other sites with 'animal housing' type => no run",
+            {
+                "completeness": {"@type": "Completeness", "animalPopulation": True},
+                "site": {"@type": "Site", "siteType": "permanent pasture"},
+                "otherSites": [{"@type": "Site", "siteType": "pond"}]
+            },
+            False
+        ),
+        (
+            "no site or other sites with 'animal housing' type and duration => no run",
+            {
+                "completeness": {"@type": "Completeness", "animalPopulation": True},
+                "site": {"@type": "Site", "siteType": "animal housing"},
+                "otherSites": [{"@type": "Site", "siteType": "pond"}],
+                "siteDuration": ""
+            },
+            False
+        ),
+        (
+            "no animal values => no run",
+            {
+                "completeness": {"@type": "Completeness", "animalPopulation": True},
+                "site": {"@type": "Site", "siteType": "animal housing"},
+                "otherSites": [{"@type": "Site", "siteType": "pond"}],
+                "siteDuration": 150,
+                "animals": [
+                    {"value": ""}
+                ]
+            },
+            False
+        ),
+        (
+            "animal values reference period is not 'average' => no run",
+            {
+                "completeness": {"@type": "Completeness", "animalPopulation": True},
+                "site": {"@type": "Site", "siteType": "animal housing"},
+                "otherSites": [{"@type": "Site", "siteType": "pond"}],
+                "siteDuration": 150,
+                "animals": [
+                    {
+                        "@type": "Animal",
+                        "value": 1
+                    },
+                    {
+                        "@type": "Animal",
+                        "value": 2,
+                        "referencePeriod": ""
+                    },
+                    {
+                        "@type": "Animal",
+                        "value": 3,
+                        "referencePeriod": "start of Cycle"
+                    },
+                ]
+            },
+            False
+        ),
+        (
+            "missing otherSitesDuration => no run",
+            {
+                "completeness": {"@type": "Completeness", "animalPopulation": True},
+                "site": {"@type": "Site", "siteType": "animal housing"},
+                "otherSites": [{"@type": "Site", "siteType": "pond"}],
+                "siteDuration": 150,
+                "animals": [
+                    {
+                        "@type": "Animal",
+                        "value": 2,
+                        "referencePeriod": ""
+                    },
+                    {
+                        "@type": "Animal",
+                        "value": 3,
+                        "referencePeriod": "average"
+                    },
+                ]
+            },
+            False
+        ),
+        (
+            "all requirements met with otherSites => run",
+            {
+                "completeness": {"@type": "Completeness", "animalPopulation": True},
+                "site": {"@type": "Site", "siteType": "pond"},
+                "siteDuration": "",
+                "otherSites": [{"@type": "Site", "siteType": "animal housing"}],
+                "otherSitesDuration": [200],
+                "animals": [
+                    {
+                        "@type": "Animal",
+                        "value": 2,
+                        "referencePeriod": ""
+                    },
+                    {
+                        "@type": "Animal",
+                        "value": 3,
+                        "referencePeriod": "average"
+                    },
+                ]
+            },
+            True
+        ),
+
+    ]
+)
+def test_should_run_animal(test_name, cycle, expected_should_run):
+    should_run, *args = should_run_animal(cycle, 'model', 'term', 'tier')
+    assert should_run == expected_should_run, test_name

tests/models/geospatialDatabase/test_croppingIntensity.py

@@ -9,14 +9,16 @@ fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 
 @patch(f"{class_path}.is_plantation")
-def test_should_run(mock_is_plantation):
+def test_should_run(mock_is_plantation, *args):
     cycle = {'@type': 'Cycle'}
 
     mock_is_plantation.return_value = False
-    assert _should_run(cycle) is True
+    should_run = _should_run(cycle)
+    assert should_run is True
 
     mock_is_plantation.return_value = True
-    assert not _should_run(cycle)
+    should_run = _should_run(cycle)
+    assert not should_run
 
 
 @patch(f"{class_path}._new_practice", side_effect=fake_new_practice)

tests/models/geospatialDatabase/test_longFallowRatio.py

@@ -9,14 +9,16 @@ fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
 
 
 @patch(f"{class_path}.is_plantation")
-def test_should_run(mock_is_plantation):
+def test_should_run(mock_is_plantation, *args):
     cycle = {'@type': 'Cycle'}
 
     mock_is_plantation.return_value = False
-    assert _should_run(cycle) is True
+    should_run = _should_run(cycle)
+    assert should_run is True
 
     mock_is_plantation.return_value = True
-    assert not _should_run(cycle)
+    should_run = _should_run(cycle)
+    assert not should_run
 
 
 @patch(f"{class_path}._new_practice", side_effect=fake_new_practice)

tests/models/site/test_management.py

@@ -297,6 +297,7 @@ def test_should_run(mock_related_cycles, *args):
 @pytest.mark.parametrize(
     "test_name,fixture_path",
     [
+        ("Products and practices", f"{fixtures_folder}"),
         ("Example 1", f"{fixtures_folder}/inputs/example1"),
         ("Example 2", f"{fixtures_folder}/inputs/example2"),
         ("Example 3", f"{fixtures_folder}/inputs/example3"),