make test for get_flat_columns