kids-math-generator.py

MAX = 50

def solve(val: int, op: str) -> int:
    if op == '+':
        a = random.randint(0, val)
        return val - a, a
    elif op == '-':
        a = random.randint(0, MAX - val)
        return val + a, a
    else:
        raise ValueError(f'Wrong operation is provided: {op}')
        
def gen_example(n=1):
     for _ in range(0, n):
         val = random.randint(0, MAX)
         op = '+' if random.randint(0, 1) == 1 else '-'
         yield (val, op, *solve(val, op))

[x for x in gen_example(5)]