diff --git a/tests/click_tests/recovery.py b/tests/click_tests/recovery.py index 143c469fba..c09046b85c 100644 --- a/tests/click_tests/recovery.py +++ b/tests/click_tests/recovery.py @@ -106,7 +106,6 @@ def select_number_of_words( raise ValueError("Invalid num_of_words") return debug.click(coords) - if debug.layout_type is LayoutType.TT: TR.assert_equals(debug.read_layout().text_content(), "recovery__num_of_words") layout = select_tt() diff --git a/tests/click_tests/test_backup_slip39_custom.py b/tests/click_tests/test_backup_slip39_custom.py index 654d9db2fd..b31cf32cfb 100644 --- a/tests/click_tests/test_backup_slip39_custom.py +++ b/tests/click_tests/test_backup_slip39_custom.py @@ -21,8 +21,8 @@ import pytest from trezorlib import device, messages from trezorlib.debuglink import LayoutType -from ..common import EXTERNAL_ENTROPY, WITH_MOCK_URANDOM, generate_entropy from .. import translations as TR +from ..common import EXTERNAL_ENTROPY, WITH_MOCK_URANDOM, generate_entropy from . import reset if TYPE_CHECKING: @@ -74,12 +74,16 @@ def test_backup_slip39_custom( groups=[(share_threshold, share_count)], ) - # confirm backup configuration if share_count > 1: - TR.assert_template(debug.read_layout().text_content(), "reset__create_x_of_y_multi_share_backup_template") + TR.assert_template( + debug.read_layout().text_content(), + "reset__create_x_of_y_multi_share_backup_template", + ) else: - TR.assert_template(debug.read_layout().text_content(), "backup__info_single_share_backup") + TR.assert_template( + debug.read_layout().text_content(), "backup__info_single_share_backup" + ) reset.confirm_read(debug) # confirm backup intro diff --git a/tests/click_tests/test_reset_bip39.py b/tests/click_tests/test_reset_bip39.py index 81af5d49b9..a3197e94d0 100644 --- a/tests/click_tests/test_reset_bip39.py +++ b/tests/click_tests/test_reset_bip39.py @@ -20,8 +20,8 @@ import pytest from trezorlib import device, messages -from ..common import WITH_MOCK_URANDOM from .. import translations as TR +from ..common import WITH_MOCK_URANDOM from . import reset from .common import go_next diff --git a/tests/input_flows_helpers.py b/tests/input_flows_helpers.py index 58b04e3496..031476937e 100644 --- a/tests/input_flows_helpers.py +++ b/tests/input_flows_helpers.py @@ -527,7 +527,9 @@ class EthereumFlow: if self.client.layout_type is LayoutType.TT: # confirm intro if info: - self.debug.click(buttons.CORNER_BUTTON, ) + self.debug.click( + buttons.CORNER_BUTTON, + ) TR.assert_equals_multiple( self.debug.read_layout().title(), [