Added test for submenu inherit for button check

ppizarror · Oct 13, 2024 · ab372a6 · ab372a6
1 parent 0346a66
commit ab372a6
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/test_menu.py b/test/test_menu.py
@@ -2634,3 +2634,17 @@ def btn() -> None:
         # Mainloop also updates last surface
         menu.mainloop(disable_loop=True)
         self.assertEqual(menu._surface_last, menu_surface)
+
+    def test_inheritance(self) -> None:
+        """
+        Test Menu class inheritance.
+        We check widgets within inherited class. #486
+        """
+
+        class SubMenu(pygame_menu.Menu):
+            def __init__(self) -> None:
+                super().__init__(title='Test', width=150, height=200, theme=pygame_menu.themes.THEME_DARK.copy())
+                help_menu = pygame_menu.Menu(title='Help', width=150, height=200)
+                self.add.button(help_menu.get_title(), help_menu)
+
+        self.assertEqual(len(SubMenu().get_widgets()), 1)