| 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() |
|
|
