File size: 1,044 Bytes
62575a2 40861f1 62575a2 40861f1 62575a2 40861f1 62575a2 40861f1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import pytest
from unittest.mock import Mock, PropertyMock, patch
from deep_translator.deepl import DeepL
from deep_translator.exceptions import AuthorizationException
@patch('deep_translator.deepl.requests')
def test_simple_translation(mock_requests):
translator = DeepL(api_key='imagine-this-is-an-valid-api-key', source='en', target='es')
# Set the request response mock.
mock_response = Mock()
mock_response.status_code = 200
mock_response.json.return_value = {
"translations": [{
"text": "hola"
}]
}
mock_requests.get.return_value = mock_response
translation = translator.translate('hello')
assert translation == 'hola'
@patch('deep_translator.deepl.requests.get')
def test_wrong_api_key(mock_requests):
translator = DeepL(api_key='this-is-a-wrong-api-key!', source="en", target="es")
# Set the response status_code only.
mock_requests.return_value = Mock(status_code=403)
with pytest.raises(AuthorizationException):
translator.translate('Hello')
|