def test_best_engine_by_language_falls_back_to_default_when_to_language_or_any_does_not_exist_in_to_preferences():
ordering = list(ENGINE_NAME_MAP.keys())
preferences = {"xx": {"xx": ordering,}, "en": {"fr": ["papago"]}}
name_map_copy = ENGINE_NAME_MAP.copy()
best_engine = _best_engine_by_language(
engines=name_map_copy,
from_language="en",
to_language="es",
language_preferences=preferences,
base_ordering=ordering,
)
assert ENGINE_NAME_MAP["microsoft"] == best_engine
def test_best_engine_by_language_follows_from_language_preferences_when_specified():
ordering = list(ENGINE_NAME_MAP.keys())
preferences = {"xx": {"xx": ordering,}, "en": {"xx": list(reversed(ordering))}}
name_map_copy = ENGINE_NAME_MAP.copy()
best_engine = _best_engine_by_language(
engines=name_map_copy,
from_language="en",
to_language="es",
language_preferences=preferences,
base_ordering=ordering,
)
assert ENGINE_NAME_MAP["yandex"] == best_engine
def test_best_engine_by_language_with_reduced_engines_dict():
ordering = list(ENGINE_NAME_MAP.keys())
preferences = {"xx": {"xx": ordering}}
name_map_copy = ENGINE_NAME_MAP.copy()
del name_map_copy["microsoft"]
best_engine = _best_engine_by_language(
engines=name_map_copy,
from_language=None,
to_language="en",
language_preferences=preferences,
base_ordering=ordering,
)
assert ENGINE_NAME_MAP["google"] == best_engine
def test_best_engine_by_language_falls_back_to_default_ordering_when_to_language_preferences_engines_are_not_available():
ordering = list(ENGINE_NAME_MAP.keys())
preferences = {"xx": {"xx": ordering, "en": ["deepl"]}}
name_map_copy = ENGINE_NAME_MAP.copy()
del name_map_copy["deepl"]
best_engine = _best_engine_by_language(
engines=name_map_copy,
from_language=None,
to_language="en",
language_preferences=preferences,
base_ordering=ordering,
)
assert ENGINE_NAME_MAP["microsoft"] == best_engine
def test_best_engine_by_language_falls_back_to_default_to_lang_preferences():
ordering = list(ENGINE_NAME_MAP.keys())
preferences = {
"xx": {"xx": ordering,},
"kr": {"xx": list(reversed(ordering)), "fr": ["papago"]},
}
name_map_copy = ENGINE_NAME_MAP.copy()
del name_map_copy["papago"]
best_engine = _best_engine_by_language(
engines=name_map_copy,
from_language="kr",
to_language="fr",
language_preferences=preferences,
base_ordering=ordering,
)
assert ENGINE_NAME_MAP["yandex"] == best_engine
def test_best_engine_by_language_with_only_default():
ordering = list(ENGINE_NAME_MAP.keys())
preferences = {"xx": {"xx": ordering}}
best_engine = _best_engine_by_language(
engines=ENGINE_NAME_MAP,
from_language=None,
to_language="en",
language_preferences=preferences,
base_ordering=ordering,
)
assert ENGINE_NAME_MAP["microsoft"] == best_engine
def test_best_engine_by_language_uses_preferences_when_both_to_and_from_specified_and_available():
ordering = list(ENGINE_NAME_MAP.keys())
preferences = {"xx": {"xx": ordering,}, "kr": {"fr": ["papago"]}}
best_engine = _best_engine_by_language(
engines=ENGINE_NAME_MAP,
from_language="kr",
to_language="fr",
language_preferences=preferences,
base_ordering=ordering,
)
assert ENGINE_NAME_MAP["papago"] == best_engine
def test_best_engine_by_language_follows_to_language_preferences_when_specified():
ordering = list(ENGINE_NAME_MAP.keys())
preferences = {"xx": {"xx": ordering, "en": ["deepl"]}}
best_engine = _best_engine_by_language(
engines=ENGINE_NAME_MAP,
from_language=None,
to_language="en",
language_preferences=preferences,
base_ordering=ordering,
)
assert ENGINE_NAME_MAP["deepl"] == best_engine
def test_best_md_table():
writer = pytablewriter.MarkdownTableWriter()
writer.headers = ["From", "To", "Engine Ordering"]
all_specified_languages = {k for k in default_language_preferences.keys()}.union(
{k for v in default_language_preferences.values() for k in v.keys()}
)
writer.value_matrix = []
for from_lang, to_lang in product(all_specified_languages, repeat=2):
if from_lang == to_lang and from_lang != "xx":
continue
name_map_copy = ENGINE_NAME_MAP.copy()
best_list = []
while len(name_map_copy) != 0:
print(name_map_copy)
best = _best_engine_by_language(
engines=name_map_copy,
from_language=from_lang,
to_language=to_lang,
language_preferences=default_language_preferences,
base_ordering=default_ordering,
)
del name_map_copy[best.NAME]
best_list.append(best.NAME)
writer.value_matrix.append([from_lang, to_lang, ", ".join(best_list)])
writer.value_matrix.sort(key=lambda x: x[0] + x[1])
writer.write_table()
try:
# raise FileNotFoundError() # uncomment to write snapshot
with open("default_preferred.md", "r") as f:
assert writer.dumps() == f.read()
except FileNotFoundError:
with open("default_preferred.md", "w") as f:
writer.dump(f)
def test_load_language_preferences_succeeds_with_default_yaml():
preferences, default_ordering = _load_language_preferences()
assert "xx" in preferences
assert set(default_ordering) == set(ENGINE_NAME_MAP.keys())