From 18a72c09e6ad1857fc88b6e5fe036700ec1e6c31 Mon Sep 17 00:00:00 2001 From: Griatch Date: Sat, 2 Dec 2023 22:19:19 +0100 Subject: [PATCH] Further fixes to tests --- evennia/contrib/rpg/llm/tests.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/evennia/contrib/rpg/llm/tests.py b/evennia/contrib/rpg/llm/tests.py index bddaac13e9..404c144086 100644 --- a/evennia/contrib/rpg/llm/tests.py +++ b/evennia/contrib/rpg/llm/tests.py @@ -13,8 +13,11 @@ from .llm_npc import LLMNPC class TestLLMClient(BaseEvenniaTestCase): - @override_settings(LLM_PROMPT_PREFIX="You are a test bot.") - @patch("evennia.contrib.rpg.llm.llm_npc.task.deferLater") + """ + Test the LLMNPC class. + + """ + def setUp(self): self.npc = create_object(LLMNPC, key="Test NPC") self.npc.db_home = None # fix a bug in test suite @@ -24,9 +27,11 @@ class TestLLMClient(BaseEvenniaTestCase): self.npc.delete() super().tearDown() + @override_settings(LLM_PROMPT_PREFIX="You are a test bot.") + @patch("evennia.contrib.rpg.llm.llm_npc.task.deferLater") def test_npc_at_talked_to(self, mock_deferLater): """ - Test the LLMNPC class. + Test the npc's at_talked_to method. """ mock_LLMClient = Mock() self.npc.ndb.llm_client = mock_LLMClient