hestia-earth-models 0.74.17__py3-none-any.whl → 0.75.0__py3-none-any.whl

tests/models/hestia/test_soilMeasurement.py

@@ -1,156 +0,0 @@
-import os
-from unittest.mock import Mock, patch
-import json
-import pytest
-
-from tests.utils import fixtures_path, fake_new_measurement
-
-from hestia_earth.models.hestia.soilMeasurement import (
-    MODEL,
-    MODEL_KEY,
-    _get_overlap,
-    _harmonise_measurements,
-    _should_run, run
-)
-
-class_path = f"hestia_earth.models.{MODEL}.{MODEL_KEY}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/soilMeasurement"
-_folders = [d for d in os.listdir(fixtures_folder) if os.path.isdir(os.path.join(fixtures_folder, d))]
-
-
-@pytest.mark.parametrize(
-    "data,expected",
-    [
-        ((1, 10, 2, 150), 8),
-        ((1, 10, 5, 15), 5),
-        ((1, 10, 10, 15), 0),
-        ((10, 10, 0, 150), 0),
-        ((150, 155, 1, 150), 0),
-        ((20, 40, 0, 30), 10),
-        ((0, 20, 0, 50), 20),
-        ((20, 60, 0, 50), 30),
-        ((10, 20, 40, 50), 0),
-        ((10, 20, 19, 50), 1),
-    ]
-)
-def test_get_overlap(data, expected):
-    result = _get_overlap(
-        data[0], data[1], data[2], data[3]
-    )
-    assert result == expected
-
-
-@pytest.mark.parametrize(
-    "measurements_list,returns_dict,expected_value",
-    [
-        (
-            [
-                {"value": [7.5], "depthUpper": 0, "depthLower": 20},
-                {"value": [6], "depthUpper": 20, "depthLower": 40},
-            ],
-            {"depthUpper": 0, "depthLower": 30},
-            7
-        ),
-        (
-            [
-                {"value": [7.5], "depthUpper": 0, "depthLower": 20},
-                {"value": [6], "depthUpper": 20, "depthLower": 40},
-            ],
-            {"depthUpper": 0, "depthLower": 50},
-            6.75
-        ),
-    ]
-)
-def test_harmonise_measurements(measurements_list, returns_dict, expected_value):
-    actual_value = _harmonise_measurements(
-        measurements_list=measurements_list,
-        standard_depth_upper=returns_dict["depthUpper"],
-        standard_depth_lower=returns_dict["depthLower"],
-    )
-    assert actual_value == expected_value
-
-
-@pytest.mark.parametrize(
-    "test_name,site,expected_should_run",
-    [
-        (
-            "no measurement => no run",
-            {"measurements": []},
-            False
-        ),
-        (
-            "missing dates => run",
-            {
-                "measurements": [
-                    {
-                        "term": {"@id": "clayContent"},
-                        "depthUpper": 0,
-                        "depthLower": 20,
-                        "value": [0]
-                    }
-                ]
-            },
-            True
-        ),
-        (
-            "no depthUpper => no run",
-            {
-                "measurements": [
-                    {
-                        "term": {"@id": "clayContent"},
-                        "dates": ["2022-01-02"],
-                        "depthLower": 20,
-                        "value": [0]
-                    }
-                ]
-            },
-            False
-        ),
-        (
-            "missing value => not run",
-            {
-                "measurements": [
-                    {
-                        "term": {"@id": "clayContent"},
-                        "dates": ["2022-01-02"],
-                        "depthUpper": 0,
-                        "depthLower": 20
-                    }
-                ]
-            },
-            False
-        ),
-        (
-            "all fields => run",
-            {
-                "measurements": [
-                    {
-                        "term": {"@id": "clayContent"},
-                        "dates": ["2022-01-02"],
-                        "depthUpper": 0,
-                        "depthLower": 20,
-                        "value": [0]
-                    }
-                ]
-            },
-            True
-        )
-    ]
-)
-def test_should_run(test_name, site, expected_should_run):
-    should_run, *args = _should_run(site)
-    assert should_run == expected_should_run, test_name
-
-
-@pytest.mark.parametrize('folder', _folders)
-@patch(f"{class_path}._new_measurement", side_effect=fake_new_measurement)
-def test_run(mock_new_measurement: Mock, folder):
-    with open(f"{fixtures_folder}/{folder}/site.jsonld", encoding='utf-8') as f:
-        site = json.load(f)
-
-    with open(f"{fixtures_folder}/{folder}/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    result = run(site)
-    print(json.dumps(result, indent=2))
-    assert result == expected, folder

tests/models/hestia/test_stockingDensityAnimalHousingAverage.py

@@ -1,42 +0,0 @@
-from unittest.mock import patch
-import pytest
-import json
-from tests.utils import fixtures_path, fake_new_practice
-
-from hestia_earth.models.hestia.stockingDensityAnimalHousingAverage import MODEL, TERM_ID, run, _should_run
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-@pytest.mark.parametrize(
-    'test_name,cycle,expected_should_run',
-    [
-        (
-            'no live animal => no run',
-            {'animals': [{'term': {'termType': 'crop'}, 'value': 10}]},
-            False
-        ),
-        (
-            'with live animal => run',
-            {'animals': [{'term': {'termType': 'liveAnimal'}, 'value': 10}]},
-            True
-        ),
-    ]
-)
-@patch(f"{class_path}.get_lookup_value", return_value=10)
-def test_should_run(mock_lookup, test_name, cycle, expected_should_run):
-    should_run, *args = _should_run(cycle)
-    assert should_run == expected_should_run, test_name
-
-
-@patch(f"{class_path}._new_practice", side_effect=fake_new_practice)
-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)
-
-    result = run(cycle)
-    assert result == expected

tests/models/hestia/test_temperatureAnnual.py

@@ -1,32 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fixtures_path, fake_new_measurement
-
-from hestia_earth.models.hestia.temperatureAnnual import MODEL, TERM_ID, run, _should_run
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-def test_should_run():
-    # no measuremrent => no run
-    site = {'measurements': []}
-    should_run, *args = _should_run(site)
-    assert not should_run
-
-    # with measuremrent => run
-    site = {'measurements': [{'term': {'@id': 'temperatureMonthly'}}]}
-    should_run, *args = _should_run(site)
-    assert should_run is True
-
-
-@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)
-
-    value = run(site)
-    assert value == expected

tests/models/hestia/test_temperatureMonthly.py

@@ -1,32 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fixtures_path, fake_new_measurement
-
-from hestia_earth.models.hestia.temperatureMonthly import MODEL, TERM_ID, run, _should_run
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-def test_should_run():
-    # no measuremrent => no run
-    site = {'measurements': []}
-    should_run, *args = _should_run(site)
-    assert not should_run
-
-    # with measuremrent => run
-    site = {'measurements': [{'term': {'@id': 'temperatureDaily'}}]}
-    should_run, *args = _should_run(site)
-    assert should_run is True
-
-
-@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)
-
-    value = run(site)
-    assert value == expected

tests/models/hestia/test_totalNitrogenPerKgSoil.py

@@ -1,36 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fixtures_path, fake_new_measurement
-
-from hestia_earth.models.hestia.totalNitrogenPerKgSoil 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}.find_term_match")
-def test_should_run(mock_measurement):
-    mock_measurement.return_value = {'value': []}
-    should_run, *args = _should_run({})
-    assert not should_run
-
-    mock_measurement.return_value = {'value': [10]}
-    should_run, *args = _should_run({})
-    assert should_run is True
-
-    mock_measurement.return_value = {'added': ['value']}
-    should_run, *args = _should_run({})
-    assert not should_run
-
-
-@patch(f"{class_path}.get_source", return_value={})
-@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)
-
-    value = run(site)
-    assert value == expected

tests/models/hestia/test_unknownPreSeasonWaterRegime.py

@@ -1,36 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fake_new_practice, fixtures_path
-
-from hestia_earth.models.hestia.unknownPreSeasonWaterRegime import MODEL, TERM_ID, _should_run, run
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-_PRACTICE_TERM_ID = 'nonFloodedPreSeasonLessThan180Days'
-
-
-@patch(f"{class_path}.get_flooded_pre_season_terms", return_value=[_PRACTICE_TERM_ID])
-def test_should_run(*args):
-    # with the practice => no run
-    cycle = {'practices': [{'term': {'@id': _PRACTICE_TERM_ID}}]}
-    should_run = _should_run(cycle)
-    assert not should_run
-
-    # without the practice => run
-    cycle = {'practices': []}
-    should_run = _should_run(cycle)
-    assert should_run is True
-
-
-@patch(f"{class_path}.get_flooded_pre_season_terms", return_value=[_PRACTICE_TERM_ID])
-@patch(f"{class_path}._new_practice", side_effect=fake_new_practice)
-def test_run(*args):
-    with open(f"{fixtures_folder}/cycle.jsonld", encoding='utf-8') as f:
-        data = json.load(f)
-
-    with open(f"{fixtures_folder}/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    result = run(data)
-    assert result == expected

tests/models/hestia/test_waterDepth.py

@@ -1,34 +0,0 @@
-from unittest.mock import patch
-import json
-from tests.utils import fixtures_path, fake_new_measurement
-
-from hestia_earth.models.hestia.waterDepth import MODEL, TERM_ID, run
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-@patch(f"{class_path}.get_source", return_value={})
-@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)
-
-    value = run(site)
-    assert value == expected
-
-
-@patch(f"{class_path}.get_source", return_value={})
-@patch(f"{class_path}._new_measurement", side_effect=fake_new_measurement)
-def test_run_river(*args):
-    with open(f"{fixtures_folder}/river/site.jsonld", encoding='utf-8') as f:
-        site = json.load(f)
-
-    with open(f"{fixtures_folder}/river/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    value = run(site)
-    assert value == expected

tests/models/hestia/test_waterSalinity.py

@@ -1,38 +0,0 @@
-from unittest.mock import Mock, patch
-import json
-from tests.utils import fixtures_path, fake_new_measurement
-
-from hestia_earth.models.hestia.waterSalinity import MODEL, TERM_ID, run
-
-class_path = f"hestia_earth.models.{MODEL}.{TERM_ID}"
-fixtures_folder = f"{fixtures_path}/{MODEL}/{TERM_ID}"
-
-
-@patch(f"{class_path}._new_measurement", side_effect=fake_new_measurement)
-@patch(f"{class_path}.related_cycles")
-def test_run_from_products(mock_related_cycles: Mock, *args):
-    with open(f"{fixtures_folder}/products/site.jsonld", encoding='utf-8') as f:
-        site = json.load(f)
-
-    with open(f"{fixtures_folder}/products/cycles.jsonld", encoding='utf-8') as f:
-        cycles = json.load(f)
-
-    mock_related_cycles.return_value = cycles
-
-    with open(f"{fixtures_folder}/products/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    value = run(site)
-    assert value == expected
-
-
-@patch(f"{class_path}._new_measurement", side_effect=fake_new_measurement)
-def test_run_from_measurement(*args):
-    with open(f"{fixtures_folder}/measurements/site.jsonld", encoding='utf-8') as f:
-        site = json.load(f)
-
-    with open(f"{fixtures_folder}/measurements/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    value = run(site)
-    assert value == expected

tests/models/impact_assessment/post_checks/test_cycle.py

@@ -1,28 +0,0 @@
-import json
-from tests.utils import fixtures_path
-
-from hestia_earth.models.impact_assessment.post_checks.cycle import run, _should_run
-
-
-def test_should_run():
-    cycle = {}
-    impact = {'cycle': cycle}
-
-    # cycle has no @id => no run
-    assert not _should_run(impact)
-    cycle['@id'] = 'id'
-
-    # cycle has an id => run
-    assert _should_run(impact)
-
-
-def test_run():
-    # contains a full cycle
-    with open(f"{fixtures_path}/impact_assessment/complete.jsonld", encoding='utf-8') as f:
-        impact = json.load(f)
-
-    cycle = impact.get('cycle')
-    cycle['@id'] = cycle['id']
-    cycle['@type'] = cycle['type']
-    value = run(impact)
-    assert value['cycle'] == {'@type': cycle['@type'], '@id': cycle['@id']}

tests/models/impact_assessment/post_checks/test_remove_cache_fields.py

@@ -1,6 +0,0 @@
-from hestia_earth.models.impact_assessment.post_checks.remove_cache_fields import run
-
-
-def test_run():
-    impact = {'cache': {}, 'added': ['cache']}
-    assert run(impact) == {'added': []}

tests/models/impact_assessment/post_checks/test_remove_no_value.py

@@ -1,17 +0,0 @@
-from hestia_earth.models.impact_assessment.post_checks.remove_no_value import run
-
-
-def test_run():
-    impacts = [
-        {'value': 10},
-        {'value': 0},
-        {'value': None},
-        {}
-    ]
-    impact = {'impacts': impacts}
-    assert run(impact) == {
-        'impacts': [
-            {'value': 10},
-            {'value': 0}
-        ]
-    }

tests/models/impact_assessment/post_checks/test_site.py

@@ -1,28 +0,0 @@
-import json
-from tests.utils import fixtures_path
-
-from hestia_earth.models.impact_assessment.post_checks.site import run, _should_run
-
-
-def test_should_run():
-    site = {}
-    impact = {'site': site}
-
-    # site has no @id => no run
-    assert not _should_run(impact)
-    site['@id'] = 'id'
-
-    # site has an id => run
-    assert _should_run(impact)
-
-
-def test_run():
-    # contains a full site
-    with open(f"{fixtures_path}/impact_assessment/complete.jsonld", encoding='utf-8') as f:
-        impact = json.load(f)
-
-    site = impact.get('site')
-    site['@id'] = site['id']
-    site['@type'] = site['type']
-    value = run(impact)
-    assert value['site'] == {'@type': site['@type'], '@id': site['@id']}

tests/models/impact_assessment/pre_checks/test_cycle.py

@@ -1,39 +0,0 @@
-from unittest.mock import patch
-from hestia_earth.schema import SchemaType
-from tests.utils import CYCLE, SITE
-
-from hestia_earth.models.impact_assessment.pre_checks.cycle import run, _should_run
-
-class_path = 'hestia_earth.models.impact_assessment.pre_checks.cycle'
-
-
-def fake_load_calculated_node(node, type):
-    return {**CYCLE} if type == SchemaType.CYCLE else {**SITE}
-
-
-def test_should_run():
-    cycle = {}
-    impact = {'cycle': cycle}
-
-    # cycle has no @id => no run
-    assert not _should_run(impact)
-    cycle['@id'] = 'id'
-
-    # cycle has an @id => run
-    assert _should_run(impact)
-
-
-def test_run_no_cycle():
-    impact = {}
-
-    value = run(impact)
-    assert 'cycle' not in value
-
-
-@patch(f"{class_path}._load_calculated_node", side_effect=fake_load_calculated_node)
-def test_run(*args):
-    impact = {'cycle': {'@id': CYCLE['@id']}}
-
-    value = run(impact)
-    # loads the cycle and the site
-    assert value['cycle'] == {**CYCLE, 'site': SITE}

tests/models/impact_assessment/pre_checks/test_site.py

@@ -1,37 +0,0 @@
-from unittest.mock import patch
-from tests.utils import SITE
-
-from hestia_earth.models.impact_assessment.pre_checks.site import run, _should_run
-
-class_path = 'hestia_earth.models.impact_assessment.pre_checks.site'
-
-
-def fake_load_calculated_node(*args): return {**SITE}
-
-
-def test_should_run():
-    site = {}
-    impact = {'site': site}
-
-    # site has no @id => no run
-    assert not _should_run(impact)
-    site['@id'] = 'id'
-
-    # site has an @id => run
-    assert _should_run(impact)
-
-
-def test_run_no_site():
-    impact = {}
-
-    value = run(impact)
-    assert 'site' not in value
-
-
-@patch(f"{class_path}._load_calculated_node", side_effect=fake_load_calculated_node)
-def test_run(*args):
-    impact = {'site': {'@id': SITE['@id']}}
-
-    value = run(impact)
-    # loads the site and the site
-    assert value['site'] == SITE

tests/models/impact_assessment/product/test_economicValueShare.py

@@ -1,36 +0,0 @@
-import json
-from tests.utils import fixtures_path
-
-from hestia_earth.models.impact_assessment.product.economicValueShare import MODEL_KEY, _should_run, run
-
-fixtures_folder = f"{fixtures_path}/impact_assessment/product/{MODEL_KEY}"
-
-
-def test_should_run():
-    # no cycle => no run
-    impact = {}
-    should_run, *args = _should_run(impact)
-    assert not should_run
-
-    # with cycle no products => no run
-    cycle = {'products': []}
-    impact['cycle'] = cycle
-    should_run, *args = _should_run(impact)
-    assert not should_run
-
-    # with product
-    product = {'term': {'@id': 'term'}}
-    cycle['products'].append(product)
-    should_run, *args = _should_run(impact)
-    assert should_run is True
-
-
-def test_run():
-    with open(f"{fixtures_folder}/impact-assessment.jsonld", encoding='utf-8') as f:
-        impact = json.load(f)
-
-    with open(f"{fixtures_folder}/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    value = run(impact)
-    assert value == expected

tests/models/impact_assessment/product/test_value.py

@@ -1,36 +0,0 @@
-import json
-from tests.utils import fixtures_path
-
-from hestia_earth.models.impact_assessment.product.value import MODEL_KEY, _should_run, run
-
-fixtures_folder = f"{fixtures_path}/impact_assessment/product/{MODEL_KEY}"
-
-
-def test_should_run():
-    # no cycle => no run
-    impact = {}
-    should_run, *args = _should_run(impact)
-    assert not should_run
-
-    # with cycle no products => no run
-    cycle = {'products': []}
-    impact['cycle'] = cycle
-    should_run, *args = _should_run(impact)
-    assert not should_run
-
-    # with product
-    product = {'term': {'@id': 'term'}}
-    cycle['products'].append(product)
-    should_run, *args = _should_run(impact)
-    assert should_run is True
-
-
-def test_run():
-    with open(f"{fixtures_folder}/impact-assessment.jsonld", encoding='utf-8') as f:
-        impact = json.load(f)
-
-    with open(f"{fixtures_folder}/result.jsonld", encoding='utf-8') as f:
-        expected = json.load(f)
-
-    value = run(impact)
-    assert value == expected