MpakoImageAIbot/tests/test_settings.py

28 lines
712 B
Python

from __future__ import annotations
from bot import MpakoBot
def test_quality_keyboard_marks_standard() -> None:
keyboard = MpakoBot._quality_keyboard("standard")
rows = keyboard.inline_keyboard
assert [row[0].text for row in rows] == [
"Стандартная ✅",
"Улучшенная",
]
assert [row[0].callback_data for row in rows] == [
"quality:standard",
"quality:improved",
]
def test_quality_keyboard_marks_improved() -> None:
keyboard = MpakoBot._quality_keyboard("improved")
rows = keyboard.inline_keyboard
assert [row[0].text for row in rows] == [
"Стандартная",
"Улучшенная ✅",
]