rafaelcaricio/LibreTranslate
1
0
Fork 0
forked from mirrors/LibreTranslate
Code Issues Pull requests Projects Releases Packages Wiki Activity

load only en & es models (avoid to download all models)

Browse source
This commit is contained in:
Sébastien Thuret 2022-04-05 16:39:31 +02:00
parent 55b05992b5
commit 7113a30a0c
No known key found for this signature in database
GPG key ID: 230B93BBA62CB043
2 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_api/conftest.py
View file

@ -2,12 +2,14 @@ import sys
import pytest import pytest
from app.app import create_app from app.app import create_app
from app.default_values import DEFAULT_ARGUMENTS
from app.main import get_args from app.main import get_args
@pytest.fixture() @pytest.fixture()
def app(): def app():
sys.argv = [''] sys.argv = ['']
DEFAULT_ARGUMENTS['LOAD_ONLY'] = "en,es"
app = create_app(get_args()) app = create_app(get_args())
yield app yield app

4
tests/test_init.py
View file

@ -4,6 +4,6 @@ from argostranslate import package
def test_boot_argos(): def test_boot_argos():
"""Test Argos translate models initialization""" """Test Argos translate models initialization"""
boot() boot(["en", "es"])
assert len(package.get_installed_packages()) > 2 assert len(package.get_installed_packages()) >= 2