def test_prompt_yesno_user_input_invalid(self): ui = UI(None, logging.DEBUG) with mock.patch('datarobot_batch_scoring.utils.input') as m_input: m_input.side_effect = ['invalid', 'yes'] assert ui.prompt_yesno('msg') m_input.assert_has_calls([mock.call('msg (Yes/No)> '), mock.call('Please type (Yes/No)> ')])
def test_prompt_yesno_user_input_n(self): ui = UI(None, logging.DEBUG) with mock.patch('datarobot_batch_scoring.utils.input') as m_input: m_input.return_value = 'n' assert not ui.prompt_yesno('msg') m_input.assert_called_with('msg (Yes/No)> ')
def test_prompt_yesno_always_no(self): ui = UI(False, logging.DEBUG) with mock.patch('datarobot_batch_scoring.utils.input') as m_input: assert not ui.prompt_yesno('msg') assert not m_input.called