Функция тестирования в python, которую нужно получить от пользователя и распечатать, используя pycharm и pytest

Question

Я написал код в python 3 и хочу написать тесты для моего кода, используя платформу Pytest. У меня есть одна функция (основная функция), которая получает ввод от пользователя (имя файла), обрабатывает его и выводит результат на экран (ответ может быть истинным или ложным). Как я могу написать тестовую функцию, которая сама вводит данные (для основного ввода) и проверяет, что выводится на консоль? (Я использую Pycharm)

Answer 1

def main():
    file_input = input(str('path to your file here'))
    with open(file_input, mode="r") as f:
        for line in f:
            reader = line.split()
            # Convert to boolean <-- Not working?
            flag = bool(reader[0])

            if flag:
                print('flag == True')
            else:
                print('flag == False')


if __name__ == '__main__':
    main()