def test_get_account(print_mock, input_mock, build_account): input_mock.sid_effect = ['5', '1'] atm = ATM() account = build_account() atm.accounts.append(account) assert atm._get_account() == account