def check_partition_column(partition_column, cols):
    """ Check partition_column existence and type

    Args:
        partition_column: partition_column name
        cols: dict with columns names and python types

    Returns:
        None
    """
    for k, v in cols.items():
        if k == partition_column:
            if v == "int":
                return
            else:
                raise InvalidPartitionColumn(
                    "partition_column must be int, and not {0}".format(v)
                )
    raise InvalidPartitionColumn(
        "partition_column {0} not found in the query".format(partition_column)
    )
Check partition_column existence and type

    Args:
        partition_column: partition_column name
        cols: dict with columns names and python types

    Returns:
        None