make tests more robust (#5747)

This commit is contained in:
Stefano Fiorucci 2023-09-08 15:50:56 +02:00 committed by GitHub
parent b5987a6d8d
commit d860a5c604
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View File

@ -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():

View File

@ -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()