mathtestsoni.py

import random

def getOp():
    op = random.randrange(1,5)

    match op:
        case 1:
            return '+'
        case 2:
            return '-'
        case 3:
            return '*'
        case 4:
            return '/'

def solve(n1,n2,op):
    if op == '+':
        return n1+n2
    elif op == '-':
        return n1-n2
    elif op == '*':
        return n1*n2
    elif op == '/':
        return round(n1/n2, 1)

def askQuestion(q,s):
    ans = float(input(q))
    if ans == s:
        return True
    else:
        return False

def main():
    numWrong = 0
    numCorrect = 0
    while (numWrong < 3):
        a = random.randrange(1,10)
        b = random.randrange(1,10)
        oper = getOp()

        solution = solve(a,b,oper)
        question = f'{a} {oper} {b} = '
        result = askQuestion(question,solution)

        if result == True:
            numCorrect = numCorrect + 1
            print('Correct')
        else:
            print('Incorrect')
            numWrong = numWrong + 1

    print(f'You got {numCorrect} correct')
main()