Simulating Python Inputs in multi-input functions using unittest

Question

I would like to simulate inputs from the user in unittest using Python 3.8. I have a function that first asks the user which program to execute and then multiple other inputs to get the required values for the said app. I would like to simulate these inputs (input()) in unittest. I was unable to find an answer from this post, as that answer uses "inputted" text and then plugs it into a function, and does not work seamlessly with input(). I would like a solution that works seamlessly with input(), as if a human was running the program, and returns the values that the function in the program outputs. Using separate functions is very tedious and would mean updating the program twice, which is not ideal. If it the only way, I am willing to deal with it, but I would prefer not to. Here is some code that will need to be tested.

main.py:

import numworksLibs

def page1():
    if (prgrmchoice == "1"):
        numer = int(input("Enter the Numerator of the Fraction: "))
        denom = int(input("Enter the Denominator of the Fraction: "))
        numworksLibs.simplify_fraction(numer, denom)

And the library file takes this input and spits out an answer (numworksLibs.py).

You may check the other answers in the question you mentioned - only the first answer uses a separate function. — MrBean Bremen
– MrBean Bremen, Commented Sep 24, 2020 at 17:22
You'll probably have to use some patch decorator, like this answer explains. I would definitely argue that you should split the function though. Lots of things can go wrong with keyboard IO, and writing unit tests for a function that receives just strings is gonna be way more straightforward and effective than a function that obtains user input and process it. — Mikael
– Mikael, Commented Sep 24, 2020 at 17:27
@Mikael I'll definitely think about this. Thanks for your input. — Revvz
– Revvz, Commented Sep 24, 2020 at 18:14
@MrBeanBremen The others weren't much help, but I'll try and adapt the program to fit the other answer like Mikael mentioned. — Revvz
– Revvz, Commented Sep 24, 2020 at 18:15
Check also this answer I gave to a somewhat related question recently. Generally, it always helps to show some minimal code, so we could give more concrete help. — MrBean Bremen
– MrBean Bremen, Commented Sep 24, 2020 at 18:23

MrBean Bremen · Accepted Answer · 2020-09-24 20:09:09Z

0

I'm not sure what exactly you want to test (maybe the output that numworksLibs generates), but as this is about mocking the input, I'll show a simplified example that does not use unknown variables or functions:

main.py

def page1():
    number = int(input("Enter the Numerator of the Fraction: "))
    denom = int(input("Enter the Denominator of the Fraction: "))
    return number, denom

test_main.py

from unittest import mock
from main import page1

@mock.patch("main.input")
def test_input(mocked_input):
    mocked_input.side_effect = ['20', '10']
    assert page1() == (20, 10)

You can put as many input values into the side_effect array as you want - this will simulate the return values of separate calls to input. You have to adapt your test code to your real code, of course.

This assumes pytest, for unittest it would look the same accept for the added self parameter.

answered Sep 24, 2020 at 20:09

MrBean Bremen

17.1k3 gold badges35 silver badges58 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Revvz Over a year ago

Worked Wonders! Thank you!

Collectives™ on Stack Overflow

Simulating Python Inputs in multi-input functions using unittest

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related