def test_load_from_long_incorrect_format(tmpdir):
"""Test for loading from long with incorrect format."""
with pytest.raises(ValueError):
dataframe = generate_example_long_table()
dataframe.drop(dataframe.columns[[3]], axis=1, inplace=True)
dataframe_path = tmpdir.join("data.csv")
dataframe.to_csv(dataframe_path, index=False)
load_from_long_to_dataframe(dataframe_path)
def test_load_from_long_to_dataframe(tmpdir):
"""Test for loading from long to dataframe."""
# create and save a example long-format file to csv
test_dataframe = generate_example_long_table()
dataframe_path = tmpdir.join("data.csv")
test_dataframe.to_csv(dataframe_path, index=False)
# load and convert the csv to sktime-formatted data
nested_dataframe = load_from_long_to_dataframe(dataframe_path)
assert isinstance(nested_dataframe, pd.DataFrame)
def test_from_long_to_nested(n_instances, n_columns, n_timepoints):
"""Test from_long_to_nested for correctness."""
X_long = generate_example_long_table(
num_cases=n_instances, series_len=n_timepoints, num_dims=n_columns
)
nested_df = from_long_to_nested(X_long)
assert is_nested_dataframe(nested_df)
assert nested_df.shape == (n_instances, n_columns)