test: Fix failing tests

Author: BriannaDelgado

CHANGELOG.md

@@ -5,6 +5,9 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 ## [Unreleased]
+### Changed
+- Updated three tests to be less flakey and reflect new behavior regarding the model type.
+
 ### Security
 - Updated dependencies to fix CVE-2025-66418 and CVE-2025-66471
 

examples/mustache/mustache.py

@@ -67,7 +67,7 @@ def convert_mustache_to_xml(
         if tag_type == "tag":
             tag_id = str(id_counter)
             id_counter += 1
-            xml += f"<{placeholder_tag} id={tag_id} />"
+            xml += f'<{placeholder_tag} id="{tag_id}" />'
             extracted_tokens[f"{placeholder_tag}#{tag_id}"] = tag_content
         else:
             xml += tag_content

tests/test_cli.py

@@ -195,7 +195,7 @@ def test_text_preserve_formatting(runner):
         main_function, 'text --to DE --preserve-formatting "proton beam"'
     )
     assert result.exit_code == 0, f"exit: {result.exit_code}\n {result.output}"
-    assert example_text["DE"].lower() in result.output
+    assert example_text["DE"].lower() in result.output.lower()
 
 
 def test_text_split_sentences(runner):

tests/test_general.py

@@ -43,7 +43,7 @@ def test_example_translation(lang, translator):
 def test_mixed_direction_text(translator):
     ar_ignore_part = "<ignore>يجب تجاهل هذا الجزء.</ignore>"
     en_sentence_with_ar_ignore_part = (
-        "<p>This is a <b>short</b> <i>sentence</i>. "
+        "<p>This is a <b>short</b> <i>sentence</i>.</p>"
         f"{ar_ignore_part} This is another sentence."
     )
     en_result = translator.translate_text(

tests/test_translate_text.py

@@ -25,14 +25,7 @@ def test_model_type(translator, model_type):
     result = translator.translate_text(
         example_text["EN"], target_lang="DE", model_type=model_type
     )
-    # TODO: use `removeprefix()` when we only support py3.8+
-    expected_model_type = str(model_type)
-    prefix_to_remove = "prefer_"
-    if expected_model_type.startswith(prefix_to_remove):
-        expected_model_type = expected_model_type[
-            len(prefix_to_remove) :  # noqa: E203
-        ]
-    assert expected_model_type == result.model_type_used
+    assert result.model_type_used is not None
 
 
 def test_string_list(translator):
@@ -213,7 +206,7 @@ def test_formality(translator, server):
         )
 
     with pytest.raises(
-        deepl.DeepLException, match=r".*formality.*target_lang.*"
+        deepl.DeepLException, match=r".*Target language.*formality.*"
     ):
         _ = translator.translate_text(
             "Test", target_lang="EN-US", formality="more"
@@ -237,7 +230,7 @@ def test_formality(translator, server):
         input_text, target_lang="TR", formality=deepl.Formality.PREFER_MORE
     )
     with pytest.raises(
-        deepl.DeepLException, match=r".*formality.*target_lang.*"
+        deepl.DeepLException, match=r".*Target language.*formality.*"
     ):
         _ = translator.translate_text(
             input_text, target_lang="TR", formality="more"