From d860a5c6042eefb39cf82f249490a0b562c12ea2 Mon Sep 17 00:00:00 2001 From: Stefano Fiorucci <44616784+anakin87@users.noreply.github.com> Date: Fri, 8 Sep 2023 15:50:56 +0200 Subject: [PATCH] make tests more robust (#5747) --- e2e/pipelines/test_standard_pipelines.py | 2 +- test/pipelines/test_standard_pipelines.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/e2e/pipelines/test_standard_pipelines.py b/e2e/pipelines/test_standard_pipelines.py index 5071bb287..fc3e96127 100644 --- a/e2e/pipelines/test_standard_pipelines.py +++ b/e2e/pipelines/test_standard_pipelines.py @@ -161,7 +161,7 @@ def test_webqa_pipeline(): assert isinstance(result, dict) assert len(result["results"]) == 1 answer = result["results"][0] - assert "Stark" in answer or "NED" in answer + assert "stark" in answer.lower() or "ned" in answer.lower() def test_faq_pipeline_batch(): diff --git a/test/pipelines/test_standard_pipelines.py b/test/pipelines/test_standard_pipelines.py index e2b6ccf5a..2d6523d7a 100644 --- a/test/pipelines/test_standard_pipelines.py +++ b/test/pipelines/test_standard_pipelines.py @@ -90,4 +90,4 @@ def test_webqa_pipeline(): assert isinstance(result, dict) assert len(result["results"]) == 1 answer = result["results"][0] - assert "Stark" in answer or "NED" in answer + assert "stark" in answer.lower() or "ned" in answer.lower()