adding test for apply_func and apply_columns