Skip to content

Commit ab372a6

Browse files
committed
Added test for submenu inherit for button check
1 parent 0346a66 commit ab372a6

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

test/test_menu.py

+14
Original file line numberDiff line numberDiff line change
@@ -2634,3 +2634,17 @@ def btn() -> None:
26342634
# Mainloop also updates last surface
26352635
menu.mainloop(disable_loop=True)
26362636
self.assertEqual(menu._surface_last, menu_surface)
2637+
2638+
def test_inheritance(self) -> None:
2639+
"""
2640+
Test Menu class inheritance.
2641+
We check widgets within inherited class. #486
2642+
"""
2643+
2644+
class SubMenu(pygame_menu.Menu):
2645+
def __init__(self) -> None:
2646+
super().__init__(title='Test', width=150, height=200, theme=pygame_menu.themes.THEME_DARK.copy())
2647+
help_menu = pygame_menu.Menu(title='Help', width=150, height=200)
2648+
self.add.button(help_menu.get_title(), help_menu)
2649+
2650+
self.assertEqual(len(SubMenu().get_widgets()), 1)

0 commit comments

Comments
 (0)