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)