def test_prompt_choice_index_error(self, prompt):
""" Out-of-range ints require user to re-input value """
invals = [44, 4, 0, -1, 3]
prompt.side_effect = lambda *_, **__: invals.pop(0)
ret = scripts.prompt_option('', ['a', 'b', 'c'])
self.assertEqual(ret, 'c')
def test_prompt_choice(self, prompt):
""" Prompt the user to choose from a list """
prompt.return_value = 2
ret = scripts.prompt_option('', ['a', 'b', 'c'])
self.assertEqual(ret, 'b')
def test_prompt_choice_bad_int(self, prompt):
""" Bad ints require user to re-input value """
invals = ['a', 'b', 1]
prompt.side_effect = lambda *_, **__: invals.pop(0)
ret = scripts.prompt_option('', ['a', 'b', 'c'])
self.assertEqual(ret, 'a')
def test_prompt_choice_bad_int(self, prompt):
""" Bad ints require user to re-input value """
invals = ["a", "b", 1]
prompt.side_effect = lambda *_, **__: invals.pop(0)
ret = scripts.prompt_option("", ["a", "b", "c"])
self.assertEqual(ret, "a")
def test_prompt_choice(self, prompt):
""" Prompt the user to choose from a list """
prompt.return_value = 2
ret = scripts.prompt_option("", ["a", "b", "c"])
self.assertEqual(ret, "b")
def test_prompt_choice_index_error(self, prompt):
""" Out-of-range ints require user to re-input value """
invals = [44, 4, 0, -1, 3]
prompt.side_effect = lambda *_, **__: invals.pop(0)
ret = scripts.prompt_option('', ['a', 'b', 'c'])
self.assertEqual(ret, 'c')
def test_prompt_choice_bad_int(self, prompt):
""" Bad ints require user to re-input value """
invals = ['a', 'b', 1]
prompt.side_effect = lambda *_, **__: invals.pop(0)
ret = scripts.prompt_option('', ['a', 'b', 'c'])
self.assertEqual(ret, 'a')
def test_prompt_choice(self, prompt):
""" Prompt the user to choose from a list """
prompt.return_value = 2
ret = scripts.prompt_option('', ['a', 'b', 'c'])
self.assertEqual(ret, 'b')
