fix: tests (match id instead of table object)

dooit-org · Nov 14, 2024 · d3e420a · d3e420a
1 parent 8157560
commit d3e420a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_ui/test_sort.py b/tests/test_ui/test_sort.py
@@ -134,7 +134,7 @@ async def test_sort_cancelled():
         await pilot.press(*list("abcd"))
         await pilot.press("escape")
 
-        current_options = [node.prompt for node in tree._options]
+        current_options = [node.id for node in tree._options]
 
         api.start_sort()
         await pilot.pause()
@@ -148,7 +148,7 @@ async def test_sort_cancelled():
         current_bar = app.bar_switcher.visible_content
         assert isinstance(current_bar, StatusBar)
 
-        new_options = [node.prompt for node in tree._options]
+        new_options = [node.id for node in tree._options]
         assert current_options == new_options