From a0eb8582c83a4127dd11e13f3e94ebef384670f9 Mon Sep 17 00:00:00 2001 From: Suhun Han Date: Sat, 23 Nov 2024 13:53:33 +0900 Subject: [PATCH] fix: test --- pytest.ini | 3 ++- tests/conftest.py | 18 +++++++++++++----- tests/test_client.py | 9 ++++----- tests/test_gtoken.py | 7 ------- 4 files changed, 19 insertions(+), 18 deletions(-) diff --git a/pytest.ini b/pytest.ini index 3f22e41..c8aa016 100644 --- a/pytest.ini +++ b/pytest.ini @@ -1,7 +1,8 @@ [pytest] -addopts = -v +addopts = -v --asyncio-mode=auto omit = tests/* +asyncio_default_fixture_loop_scope = function [run] include = googletrans/* diff --git a/tests/conftest.py b/tests/conftest.py index 4042b17..c2b2637 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,8 +1,16 @@ -from pytest import fixture +import httpx +import pytest +from googletrans import Translator, gtoken -@fixture(scope="session") -def translator(): - from googletrans import Translator - return Translator() +@pytest.fixture(scope="function") +async def translator(): + async with Translator() as t: + yield t + + +@pytest.fixture(scope="function") +async def acquirer(): + async with httpx.AsyncClient(http2=True) as client: + yield gtoken.TokenAcquirer(client=client) diff --git a/tests/test_client.py b/tests/test_client.py index 9fde14f..309af1a 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -108,24 +108,23 @@ async def test_translate_list(translator: Translator): async def test_detect_language(translator: Translator): ko = await translator.detect("한국어") en = await translator.detect("English") - rubg = await translator.detect("тест") + rubg = await translator.detect("летóво") russ = await translator.detect("привет") assert ko.lang == "ko" assert en.lang == "en" - assert rubg.lang == "mk" + assert rubg.lang == "bg" assert russ.lang == "ru" - #'bg'] @pytest.mark.asyncio async def test_detect_list(translator: Translator): - items = ["한국어", " English", "тест", "привет"] + items = ["한국어", " English", "летóво", "привет"] result = await translator.detect(items) assert result[0].lang == "ko" assert result[1].lang == "en" - assert result[2].lang == "mk" + assert result[2].lang == "bg" assert result[3].lang == "ru" diff --git a/tests/test_gtoken.py b/tests/test_gtoken.py index 37fd3b4..6efd8a7 100644 --- a/tests/test_gtoken.py +++ b/tests/test_gtoken.py @@ -1,17 +1,10 @@ from typing import Any, Callable -import httpx import pytest from googletrans import gtoken -@pytest.fixture(scope="session") -def acquirer() -> gtoken.TokenAcquirer: - client = httpx.AsyncClient(http2=True) - return gtoken.TokenAcquirer(client=client) - - @pytest.mark.asyncio async def test_acquire_token(acquirer: gtoken.TokenAcquirer) -> None: text: str = "test"