hestia-earth-models 0.51.0__py3-none-any.whl → 0.51.1__py3-none-any.whl

tests/models/schmidt2007/test_ch4ToAirWasteTreatment.py

@@ -0,0 +1,45 @@
+from unittest.mock import patch
+import json
+from tests.utils import fixtures_path, fake_new_emission
+
+from hestia_earth.models.schmidt2007.ch4ToAirWasteTreatment 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_waste_values")
+def test_should_run(mock_get_waste_values):
+    # no fuel values => no run
+    mock_get_waste_values.return_value = []
+    should_run, *args = _should_run({})
+    assert not should_run
+
+    # with fuel values => run
+    mock_get_waste_values.return_value = [0]
+    should_run, *args = _should_run({})
+    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
+
+
+@patch(f"{class_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)
+
+    with open(f"{fixtures_folder}/no-input-data-complete/result.jsonld", encoding='utf-8') as f:
+        expected = json.load(f)
+
+    value = run(cycle)
+    assert value == expected

tests/models/schmidt2007/test_utils.py

@@ -0,0 +1,39 @@
+from unittest.mock import patch
+
+from hestia_earth.models.schmidt2007.utils import get_waste_values
+
+class_path = 'hestia_earth.models.schmidt2007.utils'
+TERMS = [
+    'Oil palm mill effluent (waste)'
+]
+
+
+@patch(f"{class_path}._is_term_type_complete", return_value=True)
+def test_get_waste_values_no_inputs_complete(*args):
+    cycle = {'@type': 'Cycle', 'inputs': []}
+    assert get_waste_values('ch4ToAirWasteTreatment', cycle, '') == [0]
+
+    cycle = {'@type': 'Transformation', 'inputs': []}
+    assert get_waste_values('ch4ToAirWasteTreatment', cycle, '') == []
+
+
+@patch(f"{class_path}._is_term_type_complete", return_value=False)
+def test_get_waste_values_no_inputs_incomplete(*args):
+    cycle = {'@type': 'Cycle', 'inputs': []}
+    assert get_waste_values('ch4ToAirWasteTreatment', cycle, '') == []
+
+    cycle = {'@type': 'Transformation', 'inputs': []}
+    assert get_waste_values('ch4ToAirWasteTreatment', cycle, '') == []
+
+
+def test_get_waste_values(*args):
+    cycle = {
+        '@type': 'Cycle',
+        'products': [
+            {
+                'term': {'@id': 'oilPalmMillEffluentWaste', 'termType': 'waste'},
+                'value': [100]
+            }
+        ]
+    }
+    assert get_waste_values('ch4ToAirWasteTreatment', cycle, 'ch4EfSchmidt2007') == [1.3]