class TestCase(unittest.TestCase):
def setUp(self):
self.menu = Menu()
self.menu.add("1", "foo")
self.menu.add("2", "bar")
def test_menu_1(self):
setinput("1")
result = self.menu.show()
self.assertEqual("1", result)
def test_menu_2(self):
setinput("1")
result = self.menu.show(returns="desc")
self.assertEqual("foo", result)
def test_menu_3(self):
setinput("2")
result = self.menu.show(returns="desc")
self.assertEqual("bar", result)
def test_menu_4(self):
"""Check for regression of fix from `0.4.1`."""
m1 = Menu()
m1.add("1", "foo")
m1.add("2", "bar")
m2 = Menu()
m2.add("a", "AAA")
m2.add("b", "BBB")
m2.add("c", "CCC")
self.assertEqual(len(m1.entries), 2)
self.assertEqual(len(m2.entries), 3)
def test_menu_3(test):
"""Check for enum_menu() functionality."""
menu = Menu()
menu.add("s", "skip")
items = ["foo", "bar"]
menu = enum_menu(items, menu=menu)
setinput("1")
with test.assertRaises(EOFError):
menu.show()
setinput("s")
result = menu.show(returns="desc")
test.assertEqual("skip", result)
setinput("2")
result = menu.show(returns="desc")
test.assertEqual("foo", result)
setinput("3")
result = menu.show(returns="desc")
test.assertEqual("bar", result)
def test_menu_3(test):
"""Check for enum_menu() functionality."""
menu = Menu()
menu.add("s", "skip")
items = ["foo", "bar"]
menu = enum_menu(items, menu=menu)
setinput("1")
with test.assertRaises(EOFError):
menu.show()
setinput("s")
result = menu.show(returns="desc")
test.assertEqual("skip", result)
setinput("2")
result = menu.show(returns="desc")
test.assertEqual("foo", result)
setinput("3")
result = menu.show(returns="desc")
test.assertEqual("bar", result)
def test_menu_1(test):
"""Check for menu enum() functionality."""
menu = Menu()
menu.enum("foo")
menu.enum("bar")
setinput("1")
result = menu.show()
test.assertEqual("1", result)
setinput("1")
result = menu.show(returns="desc")
test.assertEqual("foo", result)
setinput("2")
result = menu.show(returns="desc")
test.assertEqual("bar", result)
setinput("3\n2")
result = menu.show(returns="desc")
test.assertEqual("bar", result)
def test_menu_1(test):
"""Check for menu enum() functionality."""
menu = Menu()
menu.enum("foo")
menu.enum("bar")
setinput("1")
result = menu.show()
test.assertEqual("1", result)
setinput("1")
result = menu.show(returns="desc")
test.assertEqual("foo", result)
setinput("2")
result = menu.show(returns="desc")
test.assertEqual("bar", result)
setinput("3\n2")
result = menu.show(returns="desc")
test.assertEqual("bar", result)
class TestCase(unittest.TestCase):
def setUp(self):
global TOTAL
TOTAL = 0
self.menu = Menu(inc, dec)
def test_menu_1(self):
global TOTAL
setinput("i")
result = self.menu.show()
self.assertEqual(1, TOTAL)
def test_menu_2(self):
global TOTAL
setinput("d")
result = self.menu.show()
self.assertEqual(-1, TOTAL)
def test_menu_3(self):
global TOTAL
setinput("i\ni\nd\ni\n")
result = self.menu.main(loop=True)
self.assertEqual(2, TOTAL)
import sys
from qprompt import Menu, echo
menu = Menu()
menu.add("f", "foo", lambda: echo("foo selected!"))
menu.add("b", "bar", lambda: echo("bar selected!"))
arg = None
for arg in sys.argv[1:]:
menu.run(arg)
if None == arg:
menu.show()
from qprompt import Menu
def foo():
print("foo")
def bar():
print("bar")
menu = Menu()
menu.add("f", "foo", foo)
menu.add("b", "bar", bar)
menu.add("q", "quit")
while "q" != menu.show():
pass