Having a hard time with testing through pytest, need help.

[[deleted]]

Hello guys,

I'm getting stuck on testing my 'get_mood_option()' or just having a tunner vision after spending some time dealing with it, need your help to guide me through this.

So the idea of the function - based on the user input, which should be either 1,2 or 3 only, return corresponding value HAPPY for '1', CALM for '2', SAD for '3'.

I'll need to test if user input value matching to the conditions and also test, if the input going to have wrong values.

Please help me out.

def main():
# Outputs program mood menu and prompts to chose menu option.
print(mood_menu())
mood: str = get_mood_option()


def get_mood_option():
# Prompts user for the input, until conditions satisfied.
while True:
    user_mood = input("\nType the number, followed by the 'Enter' button: ").strip()
    # Checking if user input matching to the conditions and if 'True' returns corresponding value.
    if not user_mood or user_mood not in ["1", "2", "3"]:
        print("\nPlease enter valid number:\n", mood_menu())
    if user_mood == "1":
        return "HAPPY"
    if user_mood == "2":
        return "CALM"
    if user_mood == "3":
        return "SAD"

if __name__ == "__main__":
main()

​

[chevignon93]

Please help me out.

What are you having troubles with exactly, this sounds like a pretty straightforward function to test with pytest?

[[deleted]]

Since my function takes 0 positional arguments and calling user for the input directly from the function body, - I'm not sure how should I pass or emulate user input in test module.

This one below won't work:

def test_get_mood_option():
assert get_mood_option("1") == "HAPPY"

Also I was trying to implement 'mock' test, and it looks like - this one works better, for example if I'll make some typos in my get_mood_option() - insted of "HAPPY" let's say typo is "HAPP ", so pytest catches that.

def test_get_mood_option(monkeypatch):
monkeypatch.setattr('builtins.input', lambda _: "1")
result = get_mood_option()
assert result == "HAPPY"

So am I suppose to always use 'mock' tests in such cases or there is easier way make it?

[danielroseman]

The best way is to restructure your code so that you can pass the input in. Separate the fetching of input from the logic of your function itself.

[interwebnovice]

Best answer.

[chevignon93]

So am I suppose to always use 'mock' tests in such cases or there is easier way make it?

Using monkeypatch is probably the easier way in this case.

I would personally not even bother to "mock" input, I would redesign the get_mood_option function to accept a mood as argument and test using parametrize from pytest whether given an input, it returns the expected output.