def array_of(input):
    """Return input as an array if not already."""

    if not type(input) in [list, str]:
        raise TypeError("Input must be a list or a string.")
    return [input] if isinstance(input, str) else input