update tests

MyreMylar · Feb 13, 2024 · ee9a7e0 · ee9a7e0
1 parent 9fafaff
commit ee9a7e0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_elements/test_ui_button.py b/tests/test_elements/test_ui_button.py
@@ -503,6 +503,10 @@ def test_function(data):
         button.bind(pygame_gui.UI_BUTTON_PRESSED, test_function)
         assert button._handler[pygame_gui.UI_BUTTON_PRESSED] == test_function
 
+        # test unbind
+        button.bind(pygame_gui.UI_BUTTON_PRESSED, None)
+        assert pygame_gui.UI_BUTTON_PRESSED not in button._handler
+
         button.bind(pygame_gui.UI_BUTTON_PRESSED, None)
         assert pygame_gui.UI_BUTTON_PRESSED not in button._handler