Горный программирование май 2019.py

def checkComplexNumber(complexNum):
    state = 'S0' #начальное состояние
    string = complexNum
    it = iter(string)
    isTrue = False
    while True:
        try:
            IN = next(it)
        except StopIteration:
            IN = 'EOL'
        print(state)
        if state=='S0': #состояние S0
            if IN.isdigit():
                if int(IN) >=1 and int(IN) <= 9:
                    state='S1'
                    continue
            else:
                state='SyntaxError'
                continue
        if state=='S1': #состояние S1
            if IN.isdigit():
                if int(IN) >=0 and int(IN) <= 9:
                    state='S1'
                    continue
            elif IN == '+' or IN == '-':
                state='S2'
                continue
            else:
                state='SyntaxError'
                continue
        if state=='S2':
            if IN == 'i':
                state='S3'
                continue
            else:
                state='SyntaxError'
                continue
        if state=='S3':
            if IN.isdigit():
                if int(IN) >=1 and int(IN) <= 9:
                    state='S4'
                    continue
            else:
                state='SyntaxError'
                continue
        if state=='S4':
            if IN.isdigit():
                if int(IN) >=0 and int(IN) <= 9:
                    state='S4'
                    continue
            elif IN == 'EOL':
                state='S0'
                isTrue = True
                continue
            else:
                state='SyntaxError'
                continue
        if state=='SyntaxError':
            break
    return isTrue


complexNumbers = ['123+i123','1+i1','12-i6']
complexNumbers.append(input('Enter complex to test: '))

for i in complexNumbers:
    assert checkComplexNumber(i)