Skip to content

Commit

Permalink
setup_classes & creation_date
Browse files Browse the repository at this point in the history
  • Loading branch information
lewisblake committed Sep 18, 2024
1 parent be7aeef commit 8c16602
Show file tree
Hide file tree
Showing 9 changed files with 58 additions and 18 deletions.
2 changes: 1 addition & 1 deletion pyaerocom/aeroval/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
# isort:skip_file
from .setupclasses import EvalSetup
from .setup_classes import EvalSetup
from .experiment_processor import ExperimentProcessor
2 changes: 1 addition & 1 deletion pyaerocom/aeroval/experiment_output.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@
)
from pyaerocom.aeroval.json_utils import round_floats
from pyaerocom.aeroval.modelentry import ModelEntry
from pyaerocom.aeroval.setupclasses import EvalSetup
from pyaerocom.aeroval.setup_classes import EvalSetup
from pyaerocom.aeroval.varinfo_web import VarinfoWeb
from pyaerocom.colocation.colocated_data import ColocatedData
from pyaerocom.exceptions import EntryNotAvailable, VariableDefinitionError
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from getpass import getuser
from pathlib import Path
from typing import Annotated, Literal
from datetime import datetime

from pyaerocom.aeroval.glob_defaults import VarWebInfo, VarWebScaleAndColormap

Expand Down Expand Up @@ -297,6 +298,7 @@ class ExperimentInfo(BaseModel):
public: bool = False
exp_pi: str = getuser()
pyaerocom_version: str = __version__
creation_date: str = datetime.today().strftime("%Y-%m-%d")


class EvalSetup(BaseModel):
Expand Down
8 changes: 6 additions & 2 deletions tests/aeroval/test_aeroval_HIGHLEV.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,17 @@

from pyaerocom.aeroval import ExperimentProcessor
from pyaerocom.aeroval.experiment_output import ExperimentOutput
from pyaerocom.aeroval.setupclasses import EvalSetup
from pyaerocom.aeroval.setup_classes import EvalSetup
from tests.conftest import geojson_unavail

CHK_CFG1 = {
"map": ["AERONET-Sun-od550aer_Column_TM5-AP3-CTRL-od550aer_2010.json"],
"contour": 1,
"hm": ["glob_stats_daily.json", "glob_stats_monthly.json", "glob_stats_yearly.json"],
"hm": [
"glob_stats_daily.json",
"glob_stats_monthly.json",
"glob_stats_yearly.json",
],
"hm/ts": 10, # number of .json files in sub dir
"scat": ["AERONET-Sun-od550aer_Column_TM5-AP3-CTRL-od550aer_2010.json"],
"ts": 11, # number of .json files in subdir
Expand Down
21 changes: 17 additions & 4 deletions tests/aeroval/test_experiment_output.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from pyaerocom.aeroval import ExperimentProcessor
from pyaerocom.aeroval.experiment_output import ExperimentOutput, ProjectOutput
from pyaerocom.aeroval.json_utils import read_json, write_json
from pyaerocom.aeroval.setupclasses import EvalSetup
from pyaerocom.aeroval.setup_classes import EvalSetup
from tests.conftest import geojson_unavail

BASEDIR_DEFAULT = Path(const.OUTPUTDIR) / "aeroval" / "data"
Expand Down Expand Up @@ -171,7 +171,14 @@ def test_ExperimentOutput__info_from_map_file():
"EBAS-2010-ac550aer_Surface_ECHAM-HAM-ac550dryaer_2010.json"
)

assert output == ("EBAS-2010", "ac550aer", "Surface", "ECHAM-HAM", "ac550dryaer", "2010")
assert output == (
"EBAS-2010",
"ac550aer",
"Surface",
"ECHAM-HAM",
"ac550dryaer",
"2010",
)


@pytest.mark.parametrize(
Expand Down Expand Up @@ -243,7 +250,10 @@ def test_ExperimentOutput_delete_experiment_data(tmp_path: Path, also_coldata: b
),
(
"concprcpso4",
{"colmap": "coolwarm", "scale": [0, 1.25, 2.5, 3.75, 5, 6.25, 7.5, 8.75, 10]},
{
"colmap": "coolwarm",
"scale": [0, 1.25, 2.5, 3.75, 5, 6.25, 7.5, 8.75, 10],
},
),
],
)
Expand Down Expand Up @@ -339,7 +349,10 @@ def test_ExperimentOutput_reorder_experiments_error(dummy_expout: ExperimentOutp
def test_Experiment_Output_drop_stats_and_decimals(
eval_config: dict, drop_stats, stats_decimals: int
):
eval_config["drop_stats"], eval_config["stats_decimals"] = drop_stats, stats_decimals
eval_config["drop_stats"], eval_config["stats_decimals"] = (
drop_stats,
stats_decimals,
)
cfg = EvalSetup(**eval_config)
cfg.model_cfg["mod1"] = cfg.model_cfg["TM5-AP3-CTRL"]
proc = ExperimentProcessor(cfg)
Expand Down
14 changes: 11 additions & 3 deletions tests/aeroval/test_experiment_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

from pyaerocom.aeroval.experiment_output import ExperimentOutput
from pyaerocom.aeroval.experiment_processor import ExperimentProcessor
from pyaerocom.aeroval.setupclasses import EvalSetup
from pyaerocom.aeroval.setup_classes import EvalSetup
from tests.conftest import geojson_unavail


Expand Down Expand Up @@ -35,8 +35,16 @@ def test_ExperimentProcessor_run(processor: ExperimentProcessor):
@pytest.mark.parametrize(
"cfg,kwargs,error",
[
("cfgexp2", dict(model_name="BLA"), "'No matches could be found that match input BLA'"),
("cfgexp2", dict(obs_name="BLUB"), "'No matches could be found that match input BLUB'"),
(
"cfgexp2",
dict(model_name="BLA"),
"'No matches could be found that match input BLA'",
),
(
"cfgexp2",
dict(obs_name="BLUB"),
"'No matches could be found that match input BLUB'",
),
],
)
def test_ExperimentProcessor_run_error(processor: ExperimentProcessor, kwargs: dict, error: str):
Expand Down
2 changes: 1 addition & 1 deletion tests/aeroval/test_helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
check_if_year,
make_dummy_model,
)
from pyaerocom.aeroval.setupclasses import EvalSetup
from pyaerocom.aeroval.setup_classes import EvalSetup


@pytest.mark.parametrize(
Expand Down
2 changes: 1 addition & 1 deletion tests/aeroval/test_modelmaps_engine.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytest

from pyaerocom.aeroval.modelmaps_engine import ModelMapsEngine
from pyaerocom.aeroval.setupclasses import EvalSetup
from pyaerocom.aeroval.setup_classes import EvalSetup
from pyaerocom.exceptions import ModelVarNotAvailable
from tests.fixtures.aeroval.cfg_test_exp1 import CFG

Expand Down
23 changes: 18 additions & 5 deletions tests/aeroval/test_setupclasses.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

import pytest

from pyaerocom.aeroval.setupclasses import EvalSetup
from pyaerocom.aeroval.setup_classes import EvalSetup
from pyaerocom.exceptions import EvalEntryNameError
from tests.fixtures.aeroval.cfg_test_exp1 import CFG, MODELS, OBS_GROUNDBASED

Expand Down Expand Up @@ -62,7 +62,11 @@ def test_EvalSetup_INVALID_ENTRY_NAMES(cfg_exp1: dict, error: str):


@pytest.mark.parametrize(
"update", (pytest.param(None, id="defaults"), pytest.param(dict(proj_id="blah"), id="custom"))
"update",
(
pytest.param(None, id="defaults"),
pytest.param(dict(proj_id="blah"), id="custom"),
),
)
def test_EvalSetup_ProjectInfo(eval_setup: EvalSetup, cfg_exp1: dict):
assert eval_setup.proj_info.proj_id == cfg_exp1["proj_id"]
Expand All @@ -73,12 +77,20 @@ def test_EvalSetup_ProjectInfo(eval_setup: EvalSetup, cfg_exp1: dict):
(
pytest.param(None, id="defaults"),
pytest.param(
dict(exp_id="exp42", exp_descr="Hello world!", exp_name="Lorem Ipsum...", public=True),
dict(
exp_id="exp42",
exp_descr="Hello world!",
exp_name="Lorem Ipsum...",
public=True,
),
id="custom1",
),
pytest.param(
dict(
exp_id="exp54", exp_descr="Hello world!", exp_name="Lorem Ipsum...", public=False
exp_id="exp54",
exp_descr="Hello world!",
exp_name="Lorem Ipsum...",
public=False,
),
id="custom2",
),
Expand All @@ -97,7 +109,8 @@ def test_EvalSetup_ExperimentInfo(eval_setup: EvalSetup, cfg_exp1: dict):
(
pytest.param(None, id="defaults"),
pytest.param(
dict(freqs=["yearly", "monthly"], main_freq="yearly", periods=[]), id="custom1"
dict(freqs=["yearly", "monthly"], main_freq="yearly", periods=[]),
id="custom1",
),
pytest.param(
dict(freqs=["monthly"], main_freq="monthly", periods=["2010", "2011", "2016"]),
Expand Down

0 comments on commit 8c16602

Please sign in to comment.