|
2 | 2 | # Use of this source code is governed by an MIT
|
3 | 3 | # license that can be found in the LICENSE file.
|
4 | 4 |
|
5 |
| -from .conftest import example_text |
| 5 | +from .conftest import example_text, needs_real_server |
| 6 | +from deepl.api_data import WriteResult, WritingStyle |
6 | 7 |
|
7 | 8 |
|
8 | 9 | def test_single_text(deepl_client):
|
9 | 10 | result = deepl_client.rephrase_text(
|
10 | 11 | example_text["EN"], target_lang="EN-GB"
|
11 | 12 | )
|
12 |
| - assert result.detected_source_language.upper() == "EN" |
13 |
| - epsilon = 0.2 |
14 |
| - n_original = len(example_text["EN"]) |
| 13 | + _check_sanity_of_improvements(example_text["EN"], result) |
| 14 | + |
| 15 | + |
| 16 | +@needs_real_server |
| 17 | +def test_business_style(deepl_client): |
| 18 | + input_text = "As Gregor Samsa awoke one morning from uneasy dreams he found himself transformed in his bed into a gigantic insect." # noqa |
| 19 | + result = deepl_client.rephrase_text( |
| 20 | + input_text, target_lang="EN-US", style=WritingStyle.BUSINESS.value |
| 21 | + ) |
| 22 | + _check_sanity_of_improvements(input_text, result) |
| 23 | + |
| 24 | + |
| 25 | +def _check_sanity_of_improvements( |
| 26 | + input_text: str, |
| 27 | + result: WriteResult, |
| 28 | + expected_lang_uppercase="EN", |
| 29 | + epsilon=0.2, |
| 30 | +): |
| 31 | + assert result.detected_source_language.upper() == expected_lang_uppercase |
15 | 32 | n_improved = len(result.text)
|
| 33 | + n_original = len(input_text) |
16 | 34 | assert 1 / (1.0 + epsilon) <= n_improved / n_original <= (1.0 + epsilon)
|
0 commit comments