def test_OzeEvaluation(user_name, user_password): credentials = {'user_name': user_name, 'user_password': user_password} dataset_x_path = npz_check(credentials=credentials) dataset_eval = OzeEvaluationDataset(Path(dataset_x_path.parent, 'x_test_QK7dVsy.csv'), TIME_SERIES_LENGTH, given_labels=labels) assert dataset_eval.get_x_shape() == (500, 672, 37)
def _get_dataset(user_name, user_password): return OzeNPZDataset(dataset_path=npz_check( Path('docs', 'source', 'notebooks', 'datasets'), 'dataset', credentials=_get_credentials(user_name, user_password)))
def test_download(user_name, user_password): credentials = {'user_name': user_name, 'user_password': user_password} npz_check(credentials=credentials)
def test_OzeNPZDataset(user_name, user_password): credentials = {'user_name': user_name, 'user_password': user_password} ond = OzeNPZDataset(npz_check(credentials=credentials)) assert ond.get_x_shape() == (7500, 672, 37) fd = ond.make_future_dataframe() assert fd.shape == (8000, 672, 37)