Переделайте цикл обработки команд так, чтобы для тестирования он читал команды из генератора, включая корректные и некорректные случаи, а не через `input()`.