How to test a function asking for user input in Python [duplicate]

Question

I would like to test the following Python function using pytest:

def increase_by_one()
    n = int(input())
    return n + 1

How can I write to the stdin/user input request within a function, for testing purposes?

Where possible, replace calls to input with function parameters. It makes your function easier to test, and pushes I/O towards the "edge" of your program. — chepner
– chepner, Commented Jun 25, 2020 at 12:00

Alexander Kosik · Accepted Answer · 2020-06-25 11:53:01Z

0

You can write your own input() function which will return a fix string.

save_it = input
input = lambda: 'your_simulated_input'
def test():
    s = input() # returns 'your_simulated_input'
    ...

answered Jun 25, 2020 at 11:53

Alexander Kosik

6343 silver badges10 bronze badges

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

2 Comments

chepner Over a year ago

input is in the built-in scope. You don't necessarily need to save the reference; it will always be available via builtins.input, and you can simply use del input to stop shadowing the builtin name.

Alexander Kosik Over a year ago

thank's for the hint

Collectives™ on Stack Overflow

How to test a function asking for user input in Python [duplicate]

1 Answer 1

2 Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Linked

Related