Calculate the probability of getting 100% in a test containg only singe choise quiestions

calculate-probability-of-100%.py

# this program is used to calculate the probability of getting 100% in a test containg only single choice quiestions
# x - amount of answers per question
# y - amount of questions
# formula of the function: x^0 + x^1 + x^2 + x^3 + ... + x^y

def probability(x, y):
    i = 0
    z = 0
    while(i <= y):
        z += x ** i
        i += 1
    z = f"{z:,}".replace(",", " ")
    return(f"1:{z}")

# 1st example of usage:
print(probability(int(input("How many answers per question? ")), int(input("How many questions? "))))
# 2nd example of usage:
print(probability(4, 10))
# 3rd example of usage:
x = 4
y = 10
print(probability(x, y))