awesome python program

# this is a python program

def square(num):
    return num * num

assert square(2) == 4