added unit test to generate_folds()