From d3e420a19af158df58b9e0a199eb412ec089cc38 Mon Sep 17 00:00:00 2001 From: kraanzu Date: Thu, 14 Nov 2024 18:19:04 +0530 Subject: [PATCH] fix: tests (match id instead of table object) --- tests/test_ui/test_sort.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_ui/test_sort.py b/tests/test_ui/test_sort.py index e8086fbc..62326dbd 100644 --- 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