def test_dataset_from_dict():
"""
Test ability to create a Dataset from a config element
"""
element_str = """
name: ct-23-0002
dataset:
resolution: 2T
tag_list:
- GRA-YE -23-0751X.PV
- GRA-TE -23-0698.PV
- GRA-PIT -23-0619B.PV
train_start_date: 2011-05-20T01:00:04+02:00
train_end_date: 2018-05-10T15:05:50+02:00
"""
dataset_config = get_dict_from_yaml(StringIO(element_str))["dataset"]
dataset = TimeSeriesDataset.from_dict(dataset_config.copy())
asdict = dataset.to_dict()
assert asdict["tag_list"] == [
"GRA-YE -23-0751X.PV",
"GRA-TE -23-0698.PV",
"GRA-PIT -23-0619B.PV",
]
assert asdict["resolution"] == "2T"
assert asdict["train_start_date"] == "2011-05-20T01:00:04+02:00"
assert asdict["train_end_date"] == "2018-05-10T15:05:50+02:00"
def test_dataset_from_config_checks_dates():
"""
A dataset needs to have train_start_date properly before train_end_date
"""
element_str = """
dataset:
resolution: 2T
tags:
- GRA-YE -23-0751X.PV
- GRA-TE -23-0698.PV
- GRA-PIT -23-0619B.PV
train_start_date: 2018-05-10T15:05:50+02:00
train_end_date: 2018-05-10T15:05:50+02:00
"""
dataset_config = yaml.load(element_str, Loader=yaml.FullLoader)["dataset"]
with pytest.raises(ValueError):
TimeSeriesDataset.from_dict(dataset_config)