diff --git a/tests/test_api/conftest.py b/tests/test_api/conftest.py index 3fc54aa..8ce03d1 100644 --- a/tests/test_api/conftest.py +++ b/tests/test_api/conftest.py @@ -2,12 +2,14 @@ import sys import pytest from app.app import create_app +from app.default_values import DEFAULT_ARGUMENTS from app.main import get_args @pytest.fixture() def app(): sys.argv = [''] + DEFAULT_ARGUMENTS['LOAD_ONLY'] = "en,es" app = create_app(get_args()) yield app diff --git a/tests/test_init.py b/tests/test_init.py index c1dbfe1..dacffef 100644 --- a/tests/test_init.py +++ b/tests/test_init.py @@ -4,6 +4,6 @@ from argostranslate import package def test_boot_argos(): """Test Argos translate models initialization""" - boot() + boot(["en", "es"]) - assert len(package.get_installed_packages()) > 2 + assert len(package.get_installed_packages()) >= 2