Skip to content

Commit cd0bb72

Browse files
authored
test: Stackit - make integration tests more robust (#2882)
1 parent bf0bd43 commit cd0bb72

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

integrations/stackit/tests/test_stackit_chat_generator.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,7 @@ def test_live_run(self) -> None:
248248
results = component.run(chat_messages)
249249
assert len(results["replies"]) == 1
250250
message: ChatMessage = results["replies"][0]
251-
assert "Paris" in message.text
251+
assert "paris" in message.text.lower()
252252
assert "neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8" in message.meta["model"]
253253
assert message.meta["finish_reason"] == "stop"
254254

@@ -285,13 +285,13 @@ def __call__(self, chunk: StreamingChunk) -> None:
285285

286286
assert len(results["replies"]) == 1
287287
message: ChatMessage = results["replies"][0]
288-
assert "Paris" in message.text
288+
assert "paris" in message.text.lower()
289289

290290
assert "neuralmagic/Meta-Llama-3.1-8B-Instruct-FP8" in message.meta["model"]
291291
assert message.meta["finish_reason"] == "stop"
292292

293293
assert callback.counter > 1
294-
assert "Paris" in callback.responses
294+
assert "paris" in callback.responses.lower()
295295

296296
@pytest.mark.skipif(
297297
not os.environ.get("STACKIT_API_KEY", None),
@@ -325,9 +325,9 @@ def test_live_run_with_response_format_json_schema(self):
325325
assert len(results["replies"]) == 1
326326
message: ChatMessage = results["replies"][0]
327327
msg = json.loads(message.text)
328-
assert "Paris" in msg["city"]
328+
assert "paris" in msg["city"].lower()
329329
assert isinstance(msg["country"], str)
330-
assert "France" in msg["country"]
330+
assert "france" in msg["country"].lower()
331331
assert message.meta["finish_reason"] == "stop"
332332

333333
@pytest.mark.skipif(
@@ -347,6 +347,6 @@ def test_live_run_with_response_format_pydantic_model(self, calendar_event_model
347347
assert len(results["replies"]) == 1
348348
message: ChatMessage = results["replies"][0]
349349
msg = json.loads(message.text)
350-
assert "Marketing Summit" in msg["event_name"]
350+
assert "marketing summit" in msg["event_name"].lower()
351351
assert isinstance(msg["event_date"], str)
352352
assert isinstance(msg["event_location"], str)

0 commit comments

Comments
 (0)