-
-
Save 54ny7/c6bec797d8777573b155bd78b644a3e0 to your computer and use it in GitHub Desktop.
Add default to mock_input() if no input, hide functions
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import builtins | |
| input_values = [] | |
| print_values = [] | |
| def get_display_output(): | |
| '''Get the printed output in list from the test target''' | |
| global print_values | |
| return print_values | |
| def set_keyboard_input(mocked_inputs): | |
| """Set the input in list for 'input()' | |
| ['first input for the code', 'second text', ...]""" | |
| global input_values | |
| def mock_input_output_start(): | |
| def mock_input(s=None): | |
| if s is not None: | |
| print_values.append(s) | |
| return input_values.pop(0) | |
| global input_values, print_values | |
| input_values = [] | |
| print_values = [] | |
| builtins.input = mock_input | |
| builtins.print = lambda s: print_values.append(s) | |
| mock_input_output_start() | |
| input_values = mocked_inputs |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment