def test_generate(recipe) -> None:
start = pd.Timestamp("2014-01-01", freq="D")
result = take_as_list(iterator=generate(length=10,
recipe=BASE_RECIPE,
start=start),
num=10)
assert len(result) == 10
def generate(self) -> TrainDatasets:
metadata = self.metadata
data_it = generate(
length=self.max_train_length + self.prediction_length,
recipe=self.recipe,
start=self.data_start,
)
full_length_data = take_as_list(data_it, self.num_timeseries)
test_data = [
RecipeDataset.trim_ts_item_front(
x, self.trim_length_fun(x, train_length=self.max_train_length))
for x in full_length_data
]
train_data = [
RecipeDataset.trim_ts_item_end(x, self.prediction_length)
for x in test_data
]
return TrainDatasets(
metadata=metadata,
train=ListDataset(train_data, metadata.time_granularity),
test=ListDataset(test_data, metadata.time_granularity),
)
