Skip to content

Commit 9d277db

Browse files
Update test to match 2 message format (system and user)
1 parent ee7fec2 commit 9d277db

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

formfyxer/tests/test_passive_voice_detection.py

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -264,8 +264,25 @@ def test_prompt_loading(self, mock_load_prompt):
264264
# Verify the loaded prompt was used in the API call
265265
call_args = self.mock_client.chat.completions.create.call_args
266266
messages = call_args[1]["messages"]
267-
self.assertEqual(len(messages), 1)
268-
self.assertIn("Test prompt content", messages[0]["content"])
267+
self.assertEqual(len(messages), 2)
268+
system_message = messages[0]
269+
user_message = messages[1]
270+
271+
self.assertEqual(system_message.get("role"), "system")
272+
system_content = system_message.get("content", [])
273+
self.assertTrue(system_content)
274+
self.assertIn(
275+
"Test prompt content",
276+
[part.get("text", "") for part in system_content if isinstance(part, dict)],
277+
)
278+
279+
self.assertEqual(user_message.get("role"), "user")
280+
user_content = user_message.get("content", [])
281+
self.assertTrue(user_content)
282+
self.assertIn(
283+
"Sentence: Test sentence with more words.",
284+
[part.get("text", "") for part in user_content if isinstance(part, dict)],
285+
)
269286

270287
def test_sentence_splitting(self):
271288
"""Test that text is properly split into sentences."""

0 commit comments

Comments
 (0)