寻找一种检查数学运算的方法

def check_math_operation(expression):
    # 解析表达式
    operator = expression[1]
    num1 = int(expression[0])
    num2 = int(expression[2])
    
    # 执行数学运算
    if operator == '+':
        result = num1 + num2
    elif operator == '-':
        result = num1 - num2
    elif operator == '*':
        result = num1 * num2
    elif operator == '/':
        result = num1 / num2
    else:
        return False
    
    # 检查结果是否正确
    if result == int(expression[4]):
        return True
    else:
        return False

# 测试用例
expressions = [['2', '+', '3', '=', '5'], ['4', '*', '5', '=', '20'], ['10', '-', '7', '=', '4']]

for expression in expressions:
    if check_math_operation(expression):
        print('数学运算正确')
    else:
        print('数学运算错误')