test element.delete (currently failing)

tests/test_element_delete.py

@@ -20,6 +20,7 @@ def test_remove_element_by_reference(screen: Screen):
     screen.should_not_contain('Label B')
     screen.should_contain('Label C')
     assert b.is_deleted
+    assert len(row.default_slot.children) == 2
 
 
 def test_remove_element_by_index(screen: Screen):
@@ -37,6 +38,7 @@ def test_remove_element_by_index(screen: Screen):
     screen.should_not_contain('Label B')
     screen.should_contain('Label C')
     assert b.is_deleted
+    assert len(row.default_slot.children) == 2
 
 
 def test_clear(screen: Screen):
@@ -56,6 +58,7 @@ def test_clear(screen: Screen):
     assert a.is_deleted
     assert b.is_deleted
     assert c.is_deleted
+    assert len(row.default_slot.children) == 0
 
 
 @pytest.mark.skip(reason='needs fix in element.py')  # TODO
@@ -78,3 +81,22 @@ def test_remove_parent(screen: Screen):
     assert a.is_deleted
     assert b.is_deleted
     assert c.is_deleted
+
+
+@pytest.mark.skip(reason='needs fix in element.py')  # TODO
+def test_delete_element(screen: Screen):
+    with ui.row() as row:
+        ui.label('Label A')
+        b = ui.label('Label B')
+        ui.label('Label C')
+
+    ui.button('Delete', on_click=b.delete)
+
+    screen.open('/')
+    screen.click('Delete')
+    screen.wait(0.5)
+    screen.should_contain('Label A')
+    screen.should_not_contain('Label B')
+    screen.should_contain('Label C')
+    assert b.is_deleted
+    assert len(row.default_slot.children) == 2