def test_move_up_down(self):
title = 'Please choose an option: '
options = ['option1', 'option2', 'option3']
picker = Picker(options, title)
picker.move_up()
assert picker.get_selected() == ('option3', 2)
picker.move_down()
picker.move_down()
assert picker.get_selected() == ('option2', 1)
def test_pick(self):
title = 'Please choose an option: '
options = ['option1', 'option2', 'option3']
picker = Picker(options, title)
picker.move_up()
assert picker.get_selected() == ('option3', 2)
picker.move_down()
picker.move_down()
assert picker.get_selected() == ('option2', 1)
def test_multi_select(self):
title = 'Please choose an option: '
options = ['option1', 'option2', 'option3']
picker = Picker(options, title, multi_select=True, min_selection_count=1)
assert picker.get_selected() == []
picker.mark_index()
assert picker.get_selected() == [('option1', 0)]
picker.move_down()
picker.mark_index()
assert picker.get_selected() == [('option1', 0), ('option2', 1)]
def test_options_map_func(self):
title = 'Please choose an option: '
options = [{'label': 'option1'}, {'label': 'option2'}, {'label': 'option3'}]
def get_label(option): return option.get('label')
picker = Picker(options, title, indicator='*', options_map_func=get_label)
lines, current_line = picker.get_lines()
assert lines == [title, '', '* option1', ' option2', ' option3']
assert picker.get_selected() == ({ 'label': 'option1' }, 0)
def test_default_index(self):
title = 'Please choose an option: '
options = ['option1', 'option2', 'option3']
picker = Picker(options, title, default_index=1)
assert picker.get_selected() == ('option2', 1)
def test_default_index(self):
title = 'Please choose an option: '
options = ['option1', 'option2', 'option3']
picker = Picker(options, title, default_index=1)
assert picker.get_selected() == ('option2', 1)