def get_parse_dates(file):
    """Return a list of columns for which dates have to be parsed."""
    parse_dates = [
        'click_time',
        'attributed_time'
    ]
    if 'test' in file:
        parse_dates.remove('attributed_time')
    return parse_dates