diff --git a/superscore/tests/test_page.py b/superscore/tests/test_page.py index 7380639..95aeb33 100644 --- a/superscore/tests/test_page.py +++ b/superscore/tests/test_page.py @@ -230,14 +230,14 @@ def simple_coll_return_vals(pv_name: str): ) -def test_restore_page_toggle_live(restore_page): +def test_restore_page_toggle_live(qtbot: QtBot, restore_page): tableView = restore_page.tableView live_columns = tableView.live_headers toggle_live_button = restore_page.compareLiveButton toggle_live_button.click() - assert tableView._model._poll_thread.running - assert all((not tableView.isColumnHidden(column) for column in live_columns)) + qtbot.waitUntil(lambda: tableView._model._poll_thread.running) + qtbot.waitUntil(lambda: all((not tableView.isColumnHidden(column) for column in live_columns))) toggle_live_button.click() - assert all((tableView.isColumnHidden(column) for column in live_columns)) + qtbot.waitUntil(lambda: all((tableView.isColumnHidden(column) for column in live_columns)))