PaiGram/tests/test_artifact.py

import logging

import aiofiles
import pytest
import pytest_asyncio
from flaky import flaky

from modules.apihelper.artifact import ArtifactOcrRate

LOGGER = logging.getLogger(__name__)


@pytest_asyncio.fixture
async def artifact_rate():
    _artifact_rate = ArtifactOcrRate()
    yield _artifact_rate
    await _artifact_rate.close()


# noinspection PyShadowingNames
@pytest.mark.asyncio
class TestArtifactOcrRate:
    @staticmethod
    @flaky(3, 1)
    async def test_rate_artifact(artifact_rate):
        artifact_attr = {
            "name": "翠绿的猎人之冠",
            "pos": "理之冠",
            "star": 5,
            "level": 20,
            "main_item": {"type": "cr", "name": "暴击率", "value": "31.1%"},
            "sub_item": [
                {"type": "hp", "name": "生命值", "value": "9.3%"},
                {"type": "df", "name": "防御力", "value": "46"},
                {"type": "atk", "name": "攻击力", "value": "49"},
                {"type": "cd", "name": "暴击伤害", "value": "10.9%"},
            ],
        }
        assert await artifact_rate.rate_artifact(artifact_attr)

    @staticmethod
    @flaky(3, 1)
    async def test_ocr_artifact(artifact_rate):
        async with aiofiles.open("tests/data/test_artifact.jpg", mode="rb") as f:
            photo = await f.read()
        data = await artifact_rate.get_artifact_attr(photo)
        LOGGER.info(data.text)
        assert data.status_code == 200