hestia-earth-models 0.59.5__py3-none-any.whl → 0.59.7__py3-none-any.whl

hestia_earth/models/cycle/pre_checks/startDate.py

@@ -1,52 +0,0 @@
-"""
-Pre Checks Start Date
-
-This model calculates the
-[startDate](https://hestia.earth/schema/Cycle#startDate) from the
-[endDate](https://hestia.earth/schema/Cycle#endDate) and
-[cycleDuration](https://hestia.earth/schema/Cycle#cycleDuration).
-"""
-from dateutil.relativedelta import relativedelta
-from hestia_earth.utils.date import is_in_days
-from hestia_earth.utils.tools import safe_parse_date
-
-from hestia_earth.models.log import logRequirements, logShouldRun
-from .. import MODEL
-
-REQUIREMENTS = {
-    "Cycle": {
-        "endDate": "",
-        "cycleDuration": ""
-    }
-}
-RETURNS = {
-    "Cycle": {
-        "startDate": ""
-    }
-}
-MODEL_KEY = 'startDate'
-
-
-def _run(cycle: dict):
-    days = -float(cycle.get('cycleDuration'))
-    return (safe_parse_date(cycle.get('endDate')) + relativedelta(days=days)).strftime('%Y-%m-%d')
-
-
-def _should_run(cycle: dict):
-    has_full_date = is_in_days(cycle.get('endDate'))
-    cycleDuration = cycle.get('cycleDuration')
-
-    logRequirements(cycle, model=MODEL, key=MODEL_KEY,
-                    has_full_date=has_full_date,
-                    cycleDuration=cycleDuration)
-
-    should_run = all([
-        has_full_date,
-        cycleDuration is not None,
-        cycle.get(MODEL_KEY) is None
-    ])
-    logShouldRun(cycle, MODEL, None, should_run, key=MODEL_KEY)
-    return should_run
-
-
-def run(cycle: dict): return {**cycle, MODEL_KEY: _run(cycle)} if _should_run(cycle) else cycle

tests/models/cycle/pre_checks/test_startDate.py

@@ -1,44 +0,0 @@
-import json
-from tests.utils import fixtures_path
-
-from hestia_earth.models.cycle.pre_checks.startDate import run, _should_run
-
-fixtures_folder = f"{fixtures_path}/cycle/pre_checks/startDate"
-
-
-def test_should_run():
-    cycle = {}
-
-    # no endDate => no run
-    cycle['endDate'] = None
-    assert not _should_run(cycle)
-
-    cycle['endDate'] = '2010'
-    # no cycleDuration => no run
-    cycle['cycleDuration'] = None
-    assert not _should_run(cycle)
-
-    cycle['cycleDuration'] = 120
-    # with a startDate => no run
-    cycle['startDate'] = '2010'
-    assert not _should_run(cycle)
-
-    cycle['startDate'] = None
-    # endDate not precise enough => no run
-    cycle['endDate'] = '2020-01'
-    assert not _should_run(cycle)
-
-    # endDate is precise enough => run
-    cycle['endDate'] = '2020-01-01'
-    assert _should_run(cycle) is True
-
-
-def test_run():
-    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