mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 20:10:36 +00:00
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
|
import logging
|
||
|
|
|
||
|
|
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):
|
||
|
|
with open("tests/data/test_artifact.jpg", "rb") as f:
|
||
|
|
photo = f.read()
|
||
|
|
data = await artifact_rate.get_artifact_attr(photo)
|
||
|
|
LOGGER.info(data.text)
|
||
|
|
assert data.status_code == 200
|