mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
0c62cc8b27
⚙️ 获取 Redis cookies 池函数从 `zrevrange` 替代为 `zrange` 🐛 修复 fakeredis 无效 📝优化文本描述 🎨 提高代码质量 Co-authored-by: 洛水居室 <luoshuijs@outlook.com>
44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
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
|