From 32e6f71856c786db9e546fd23564941d665449b6 Mon Sep 17 00:00:00 2001 From: Dave Pearson Date: Thu, 10 Nov 2022 20:41:11 +0000 Subject: [PATCH] Be 100% sure all removals take place --- tests/test_widget_removing.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/test_widget_removing.py b/tests/test_widget_removing.py index 62b9dca32..341866a9f 100644 --- a/tests/test_widget_removing.py +++ b/tests/test_widget_removing.py @@ -122,7 +122,9 @@ async def test_widget_remove_order(): await pilot.app.mount( Removable(Removable(Removable(id="grandchild"), id="child"), id="parent") ) + assert len(pilot.app.screen.walk_children(with_self=False)) == 3 await pilot.app.screen.children[0].remove() + assert len(pilot.app.screen.walk_children(with_self=False)) == 0 assert removals == ["grandchild", "child", "parent"] async def test_query_remove_order(): @@ -138,5 +140,7 @@ async def test_query_remove_order(): await pilot.app.mount( Removable(Removable(Removable(id="grandchild"), id="child"), id="parent") ) + assert len(pilot.app.screen.walk_children(with_self=False)) == 3 await pilot.app.query(Removable).remove() + assert len(pilot.app.screen.walk_children(with_self=False)) == 0 assert removals == ["grandchild", "child", "parent"]