Правильно ли я проверяю соединение со снежинкой в ​​своих тестах?

Question

Я устанавливаю соединение со снежинкой и хочу выполнить его модульное тестирование путем проверки соединения, однако я не уверен, правильно ли я это делаю.

В тесте я исправляю snowflake.connector.connect и вызываю метод, утверждая, что метод выполнялся с определенными параметрами.

Затем, чтобы проверить обработку ошибок, я установил побочный эффект проверяемого соединения на Exception, затем сделаю утверждение, чтобы увидеть, было ли возбуждено исключение.

Не думаю, что я делаю это правильно, потому что мое тестовое покрытие говорит, что последние две строки моего кода не покрыты. Есть идеи? Спасибо!

CONNECTION-------------
def connection():
    try:
        connection = snowflake.connector.connect(
                                account = 'account',
                                user = 'user',
                                password = 'password',
                                warehouse = 'warehouse',
                                role = 'role'
                                )

        return connection
    except Exception as error:
        print (error)

TEST---------------
@patch('snowflake.connector.connect')
def test_connection(self):

    connection = connection()

    mock_connection.assert_call_once_with(
            account = 'account',
            user = 'user',
            password = 'password',
            warehouse = 'warehouse',
            role = 'role'        
            )

    mock_connect.side_effect = Exception

    with self.assertRaises(Exception):
        connection()