def test_bind_down(self) -> None:
keyboard = keyboards.Keyboard()
test_string = 'set a'
keyboard.bind_on_press(self.a_key, lambda: self.set_a(test_string))
# function is called when you press the key
keyboards.pg.key.pressed[self.a_key] = 1
keyboard.handle_input()
self.assertEqual(self.a, test_string)
# reset to detect future calls
self.a = ''
# nothing happens if you keep it help down
keyboards.pg.key.pressed[self.a_key] = 1
keyboard.handle_input()
self.assertEqual(self.a, '')
# nothing happens when you release
keyboards.pg.key.pressed[self.a_key] = 0
keyboard.handle_input()
self.assertEqual(self.a, '')
# function called again when you press down
keyboards.pg.key.pressed[self.a_key] = 1
keyboard.handle_input()
self.assertEqual(self.a, test_string)
def test_multiple_press(self) -> None:
keyboard = keyboards.Keyboard()
n_bindings = len(keyboard._bindings)
test_string_a = 'set a'
test_string_b = 'set b'
test_string_c = 'set c'
test_mouse_1 = 'mouse 1'
test_mouse_2 = 'mouse 2'
keyboard.bind(self.a_key, lambda: self.set_a(test_string_a))
keyboard.bind(self.b_key, lambda: self.set_b(test_string_b))
keyboard.bind(self.c_key, lambda: self.set_c(test_string_c))
keyboard.bind_mouse(self.a_key, lambda: self.set_d(test_mouse_1))
keyboard.bind_mouse(self.b_key, lambda: self.set_e(test_mouse_2))
self.assertEqual(len(keyboard._bindings), 3 + n_bindings)
self.assertEqual(len(keyboard._mouse_bindings), 2)
# every combination of 3 keys and two mouse presses
for key_1, key_2, key_3, mouse_1, mouse_2 in product(*[[0, 1]] * 5):
# press the keys
keyboards.pg.key.pressed[0] = key_1
keyboards.pg.key.pressed[1] = key_2
keyboards.pg.key.pressed[2] = key_3
keyboards.pg.mouse.pressed[0] = mouse_1
keyboards.pg.mouse.pressed[1] = mouse_2
# .handle_input() combinations
keyboard.handle_input()
if key_1:
self.assertEqual(self.a, test_string_a)
if key_2:
self.assertEqual(self.b, test_string_b)
if key_3:
self.assertEqual(self.c, test_string_c)
if mouse_1:
self.assertEqual(self.d, test_mouse_1)
if mouse_2:
self.assertEqual(self.e, test_mouse_2)
keyboards.pg.key.pressed[0] = 0
keyboards.pg.key.pressed[1] = 0
keyboards.pg.key.pressed[2] = 0
keyboards.pg.mouse.pressed[0] = 0
keyboards.pg.mouse.pressed[1] = 0
keyboard.handle_input()
# reset to register the next clicks
self.a = ''
self.b = ''
self.c = ''
self.d = ''
self.e = ''
def test_mouse(self) -> None:
keyboard = keyboards.Keyboard()
test_string = 'set a'
keyboard.bind_mouse(self.a_key, lambda: self.set_a(test_string))
keyboards.pg.mouse.pressed[self.a_key] = 1
keyboard.handle_input()
self.assertEqual(self.a, test_string)
def test_bind(self) -> None:
keyboard = keyboards.Keyboard()
n_bindings = len(keyboard._bindings)
test_string = 'set a'
keyboard.bind(self.a_key, lambda: self.set_a(test_string))
keyboards.pg.key.pressed[self.a_key] = 1
# test we set the key
self.assertEqual(len(keyboard._bindings), 1 + n_bindings)
# ensure we haven't changed this yet
self.assertEqual(self.a, '')
keyboard.handle_input()
self.assertEqual(self.a, test_string)
def test_only_handle(self) -> None:
keyboard = keyboards.Keyboard()
n_bindings = len(keyboard._bindings)
test_string_a = 'set a'
test_string_b = 'set b'
test_string_c = 'set c'
keyboard.bind(self.a_key, lambda: self.set_a(test_string_a))
keyboard.bind(self.b_key, lambda: self.set_b(test_string_b))
keyboard.bind(self.c_key, lambda: self.set_c(test_string_c))
# test we set the key
self.assertEqual(len(keyboard._bindings), 3 + n_bindings)
# ensure we haven't changed this yet
self.assertEqual(self.b, '')
keyboards.pg.key.pressed[self.b_key] = 1
keyboard.handle_input()
self.assertEqual(self.b, test_string_b)
keyboards.pg.key.pressed[self.c_key] = 1
keyboard.handle_input()
self.assertEqual(self.c, test_string_c)
self.a = ''
self.b = ''
self.c = ''
keyboards.pg.key.pressed[self.a_key] = 1
keyboards.pg.key.pressed[self.b_key] = 1
keyboards.pg.key.pressed[self.c_key] = 1
keyboard.handle_input(allowed_keys=[self.a_key])
self.assertEqual(self.a, test_string_a)
self.assertEqual(self.b, '')
self.assertEqual(self.c, '')