From 0528e44d071ab34da7613750da66add5b754529a Mon Sep 17 00:00:00 2001 From: Pavel Kirilin Date: Fri, 12 Sep 2025 17:57:36 +0200 Subject: [PATCH] Fixed regex. --- lenochka/brain/tests/test_brain_cell.py | 32 ++++++++++++++++--------- lenochka/mouth.py | 2 +- values.yaml | 3 +++ 3 files changed, 25 insertions(+), 12 deletions(-) diff --git a/lenochka/brain/tests/test_brain_cell.py b/lenochka/brain/tests/test_brain_cell.py index d079756..ff70649 100644 --- a/lenochka/brain/tests/test_brain_cell.py +++ b/lenochka/brain/tests/test_brain_cell.py @@ -1,16 +1,26 @@ +from typing import Union + +import pytest + from lenochka.brain.brain_cell import BrainCell -def test_replies() -> None: +@pytest.mark.parametrize( + ("sentence", "expected_reply"), + [ + ("Удав", "Пиздав."), + ("Дакимакура", "Пиздакимакура."), + ("Никак нет", None), + ("нет", "Пидора ответ."), + ("Уважаемая, Леночка", "Какая я тебе нахуй Леночка, хуесос?"), + ("Леночка, принесите кофе", "Какая я тебе нахуй Леночка, хуесос?"), + ("Да нет Леночка", "Какая я тебе нахуй Леночка, хуесос?"), + ], +) +def test_replies(sentence: str, expected_reply: Union[str, None]) -> None: """Test reply generator.""" cell = BrainCell() - assert cell.create_reply("Удав") == "Пиздав." - assert cell.create_reply("Дакимакура") == "Пиздакимакура." - - assert cell.create_reply("Никак нет") is None - assert cell.create_reply("нет") == "Пидора ответ." - - assert cell.create_reply("Уважаемая, Леночка") == "Какая я тебе нахуй Леночка, хуесос?" - assert cell.create_reply("Леночка, принесите кофе") == "Какая я тебе нахуй Леночка, хуесос?" - - assert cell.create_reply("Да нет Леночка") == "Какая я тебе нахуй Леночка, хуесос?" + if expected_reply: + assert cell.create_reply(sentence) == expected_reply + else: + assert cell.create_reply(sentence) is None diff --git a/lenochka/mouth.py b/lenochka/mouth.py index ef61963..7b870e4 100644 --- a/lenochka/mouth.py +++ b/lenochka/mouth.py @@ -18,6 +18,6 @@ def talk_to_the_wind() -> None: ears = ConnectionToEars(brain_connection) ears.register_message_handler( brain.think, - regexp="да", + regexp="(да|нет|леночка)", ) world.start_polling(ears) diff --git a/values.yaml b/values.yaml index a473de7..8dfc2fd 100644 --- a/values.yaml +++ b/values.yaml @@ -3,6 +3,9 @@ nameOverride: "lenochka" image: tag: "latest" +updateStrategy: + type: Recreate + service: enabled: false