Spaces:
Running
Running
File size: 1,260 Bytes
63fc551 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
from modules.translation.deepl_api import DeepLAPI
from modules.translation.nllb_inference import NLLBInference
from test_config import *
import os
import pytest
@pytest.mark.parametrize("model_size, file_path", [
(TEST_NLLB_MODEL, TEST_SUBTITLE_SRT_PATH),
(TEST_NLLB_MODEL, TEST_SUBTITLE_VTT_PATH),
])
def test_nllb_inference(
model_size: str,
file_path: str
):
nllb_inferencer = NLLBInference()
print("Device : ", nllb_inferencer.device)
result_str, file_paths = nllb_inferencer.translate_file(
fileobjs=[file_path],
model_size=model_size,
src_lang="eng_Latn",
tgt_lang="kor_Hang",
)
assert isinstance(result_str, str)
assert isinstance(file_paths[0], str)
@pytest.mark.parametrize("file_path", [
TEST_SUBTITLE_SRT_PATH,
TEST_SUBTITLE_VTT_PATH,
])
def test_deepl_api(
file_path: str
):
deepl_api = DeepLAPI()
api_key = os.getenv("DEEPL_API_KEY")
result_str, file_paths = deepl_api.translate_deepl(
auth_key=api_key,
fileobjs=[file_path],
source_lang="English",
target_lang="Korean",
is_pro=False,
add_timestamp=True,
)
assert isinstance(result_str, str)
assert isinstance(file_paths[0], str)
|