Испытание модуля Python

Question

Я новичок в модульном тестировании. Мне было интересно, как вы можете тестировать код ниже с помощью теста python unit. Вы можете использовать какие-либо рамки для тестирования, но, пожалуйста, укажите, почему и как быть. На самом деле цените помощь каждого.

BTW вы можете вносить изменения в код, если он делает то, что он делает прямо сейчас. В любом случае это было BS: P

pyg = 'ay' 

original = raw_input('Enter a word:') 

word = original.lower() 

first = word[0] 

    if first == "a" and "e" and "i" and "o" and "u": 
     new_word = word + pyg 
     print new_word 
    else: 
     consonant_fix = word[1:] 
     new_word = consonant_fix + word[0] + pyg 
     print new_word 

Answer 1

Я не думаю, что это хороший кандидат для модульного тестирования.

Ввод/вывод не должен иметь значения.

Лучше реорганизовать это в функцию, которая принимает строку, работает с ней и возвращает новую строку. Затем вы можете написать модульные тесты с конкретными входами, чтобы проверить поведение, которое вы хотите.

Как написано, ваш код не поддается тестированию.

Вот как я бы переписать его:

def rearrange(original, pyg): 
    word = original.lower() 
    first = word[0] 
    if first == "a" and "e" and "i" and "o" and "u": 
     new_word = word + pyg 
    else: 
     consonant_fix = word[1:] 
     new_word = consonant_fix + word[0] + pyg 
    return new_word 


if __name__ == '__main__': 
    pyg = 'ay' 
    original = raw_input('Enter a word:') 
    print rearrange(original, pyg) 

Вы можете проверить эту rearrange() функцию с различными входами в настоящее время.