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)