Reading int input for HackerRank

read_ints.py

def read_tokens():
    return input().strip().split()


def read_ints():
    return [int(s) for s in read_tokens()]


a, = read_ints()  # 1 int
a, b, c = read_ints()  # several
a = read_ints()  # list