example.py

def my_example_function():
    print("Hi")

    name = input("Tell me your name")
    age = input("Tell me your age")

    print("Hi {}, you are {} years old".format(name, age))

example_test.py

from project.example import my_example_function
from project.test.tud_test_base import mock_input_output_start, mock_input_output_end, set_input, get_output


def test_1():
    mock_input_output_start()
    set_input(["Mauricio", 33])

    my_example_function()

    output = get_output()

    assert output == ["Hi","Tell me your name", "Tell me your age", "Hi Mauricio, you are 33 years old"]


    mock_input_output_end()

tud_test_base.py

import builtins

input_values = []
print_values = []

previous_input = None
previous_print = None


def mock_input(s):
    print_values.append(s)
    return input_values.pop(0)


def mock_input_output_start():
    global input_values, print_values, previous_input, previous_print

    input_values = []
    print_values = []

    previous_input = builtins.input
    previous_print = builtins.print

    builtins.input = mock_input
    builtins.print = lambda s: print_values.append(s)


def mock_input_output_end():
    builtins.input = previous_input
    builtins.print = previous_print


def get_output():
    global print_values
    return print_values


def set_input(mocked_inputs):
    global input_values
    input_values = mocked_inputs