hestia-earth-models 0.70.1__py3-none-any.whl → 0.70.3__py3-none-any.whl

tests/models/ipcc2019/test_co2ToAirOrganicSoilCultivation.py

@@ -0,0 +1,62 @@
+import json
+from pytest import mark
+from unittest.mock import MagicMock, patch
+
+from hestia_earth.models.ipcc2019.co2ToAirOrganicSoilCultivation import MODEL, TERM_ID, _should_run, run
+from tests.utils import fake_new_emission, fixtures_path
+
+class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+
+# subfolder, should_run
+PARAMS_SHOULD_RUN = [
+    ("acacia", True),
+    ("annual-crops", True),
+    ("grassland", True),
+    ("oil-palm", True),
+    ("other", False),
+    ("paddy-rice-cultivation", True),
+    ("perennial-crops", True),
+    ("sago-palm", True),
+    ("polar", False),
+    ("mineral-soil", True),
+    ("unknown-soil", False),
+    ("unknown-land-occupation", False)
+]
+IDS_SHOULD_RUN = [p[0] for p in PARAMS_SHOULD_RUN]
+
+
+@mark.parametrize("subfolder, expected", PARAMS_SHOULD_RUN, ids=IDS_SHOULD_RUN)
+def test_should_run(
+    subfolder: str,
+    expected: bool
+):
+    folder = f"{fixtures_folder}/{subfolder}"
+
+    with open(f"{folder}/cycle.jsonld", encoding="utf-8") as f:
+        cycle = json.load(f)
+
+    result, *_ = _should_run(cycle)
+    assert result == expected
+
+
+PARAMS_RUN = [subfolder for subfolder, should_run in PARAMS_SHOULD_RUN if should_run]
+
+
+@mark.parametrize("subfolder", PARAMS_RUN)
+@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+def test_run(
+    _mock_new_emission: MagicMock,
+    subfolder: str,
+):
+    folder = f"{fixtures_folder}/{subfolder}"
+
+    with open(f"{folder}/cycle.jsonld", encoding="utf-8") as f:
+        cycle = json.load(f)
+
+    with open(f"{folder}/result.jsonld", encoding="utf-8") as f:
+        expected = json.load(f)
+
+    result = run(cycle)
+    assert result == expected

tests/models/ipcc2019/test_co2ToAirSoilOrganicCarbonStockChange.py

@@ -59,15 +59,17 @@ RUN_SCENARIOS = [
     ("overlapping-cycles", 4),
     ("complex-overlapping-cycles", 5),
     ("missing-measurement-dates", 3),
-    ("no-organic-carbon-measurements", 1),               # Closes issue #700
-    ("non-consecutive-organic-carbon-measurements", 1),  # Closes issue #827
-    ("multiple-method-classifications", 5),              # Closes issue #764
-    ("non-soil-based-gohac-system", 3),                  # Closes issue #848
-    ("with-gapfilled-start-date-end-date", 1),           # Closes issue #972
-    ("forest-to-orchard-with-ground-cover", 3),          # Closes issue #989
-    ("orchard-data-complete", 3),                 # Closes issue #1011
-    ("orchard-data-partially-complete", 3),       # Closes issue #1011
-    ("with-linalgerror", 1)
+    ("no-organic-carbon-measurements", 1),               # Closes #700
+    ("non-consecutive-organic-carbon-measurements", 1),  # Closes #827
+    ("multiple-method-classifications", 5),              # Closes #764
+    ("non-soil-based-gohac-system", 3),                  # Closes #848
+    ("with-gapfilled-start-date-end-date", 1),           # Closes #972
+    ("forest-to-orchard-with-ground-cover", 3),          # Closes #989
+    ("orchard-data-complete", 3),                        # Closes #1011
+    ("orchard-data-partially-complete", 3),              # Closes #1011
+    ("with-linalgerror", 1),
+    ("no-measurements-data-complete", 1),                # Closes #1227
+    ("no-measurements-data-incomplete", 1)               # Closes #1227
 ]
 """List of (subfolder: str, num_cycles: int)."""
 

tests/models/ipcc2019/test_n2OToAirOrganicSoilCultivationDirect.py

@@ -0,0 +1,61 @@
+import json
+from pytest import mark
+from unittest.mock import MagicMock, patch
+
+from hestia_earth.models.ipcc2019.n2OToAirOrganicSoilCultivationDirect import MODEL, TERM_ID, _should_run, run
+from tests.utils import fake_new_emission, fixtures_path
+
+class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
+
+# subfolder, should_run
+PARAMS_SHOULD_RUN = [
+    ("acacia", True),
+    ("annual-crops", True),
+    ("grassland", True),
+    ("oil-palm", True),
+    ("other", False),
+    ("paddy-rice-cultivation", True),
+    ("perennial-crops", True),
+    ("sago-palm", True),
+    ("polar", False),
+    ("mineral-soil", True),
+    ("unknown-soil", False),
+    ("unknown-land-occupation", False)
+]
+IDS_SHOULD_RUN = [p[0] for p in PARAMS_SHOULD_RUN]
+
+
+@mark.parametrize("subfolder, expected", PARAMS_SHOULD_RUN, ids=IDS_SHOULD_RUN)
+def test_should_run(
+    subfolder: str,
+    expected: bool
+):
+    folder = f"{fixtures_folder}/{subfolder}"
+
+    with open(f"{folder}/cycle.jsonld", encoding="utf-8") as f:
+        cycle = json.load(f)
+
+    result, *_ = _should_run(cycle)
+    assert result == expected
+
+
+PARAMS_RUN = [subfolder for subfolder, should_run in PARAMS_SHOULD_RUN if should_run]
+
+
+@mark.parametrize("subfolder", PARAMS_RUN,)
+@patch(f"{class_path}._new_emission", side_effect=fake_new_emission)
+def test_run(
+    _mock_new_emission: MagicMock,
+    subfolder: str
+):
+    folder = f"{fixtures_folder}/{subfolder}"
+
+    with open(f"{folder}/cycle.jsonld", encoding="utf-8") as f:
+        cycle = json.load(f)
+
+    with open(f"{folder}/result.jsonld", encoding="utf-8") as f:
+        expected = json.load(f)
+
+    result = run(cycle)
+    assert result == expected

tests/models/site/test_grouped_measurement.py

@@ -0,0 +1,20 @@
+import json
+from unittest.mock import patch
+from tests.utils import fixtures_path, fake_new_measurement
+
+from hestia_earth.models.site.grouped_measurement import MODEL, MODEL_KEY, run
+
+class_path = f"hestia_earth.models.{MODEL}.{MODEL_KEY}"
+fixtures_folder = f"{fixtures_path}/{MODEL}/{MODEL_KEY}"
+
+
+@patch(f"{class_path}._new_measurement", side_effect=fake_new_measurement)
+def test_run(*args):
+    with open(f"{fixtures_folder}/site.jsonld", encoding='utf-8') as f:
+        site = json.load(f)
+
+    with open(f"{fixtures_folder}/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    result = run(site)
+    assert result == expected