|
import os |
|
import sys |
|
import pytest |
|
|
|
pytest_plugins = ('pytest_asyncio') |
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
|
|
|
from manga_translator.translators import ( |
|
TranslatorChain, |
|
dispatch, |
|
) |
|
|
|
@pytest.mark.asyncio |
|
async def test_specified_translator(translator, tgt_lang, text, count): |
|
if translator is None: |
|
pytest.skip() |
|
|
|
print() |
|
for i in range(count): |
|
if text is None: |
|
queries_list = [ |
|
['Hallo', '', 'English is a West Germanic language in the Indo-European language family, with its earliest forms spoken by the inhabitants of early medieval England.', 'Test case 5. HELLO THERE I WANT an audition! YOYOYOYO', '目标意识'], |
|
['僕はアイネと共に一度、宿の方に戻った', '改めて直面するのは部屋の問題――部屋のベッドが一つでは、さすがに狭すぎるだろう。'], |
|
['....DO YOU HAVE EXPERIENCE IN TAKING CARE OF SICK PEOPLE..?'], |
|
] |
|
else: |
|
queries_list = [[text]] |
|
|
|
|
|
chain = TranslatorChain(f'{translator}:{tgt_lang}') |
|
for queries in queries_list: |
|
print(queries) |
|
print('-->') |
|
print(await dispatch(chain, queries)) |
|
print() |
|
|
