Skip to content

Commit 505c8d2

Browse files
committed
tests: Add style test
1 parent 1904841 commit 505c8d2

File tree

1 file changed

+22
-4
lines changed

1 file changed

+22
-4
lines changed

tests/test_rephrase_text.py

+22-4
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,33 @@
22
# Use of this source code is governed by an MIT
33
# license that can be found in the LICENSE file.
44

5-
from .conftest import example_text
5+
from .conftest import example_text, needs_real_server
6+
from deepl.api_data import WriteResult, WritingStyle
67

78

89
def test_single_text(deepl_client):
910
result = deepl_client.rephrase_text(
1011
example_text["EN"], target_lang="EN-GB"
1112
)
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
1532
n_improved = len(result.text)
33+
n_original = len(input_text)
1634
assert 1 / (1.0 + epsilon) <= n_improved / n_original <= (1.0 + epsilon)

0 commit comments

Comments
 (0)