relax test assertion (#4229)

This commit is contained in:
Massimiliano Pippi 2023-02-22 12:37:09 +01:00 committed by GitHub
parent 1e4ef24ae9
commit 262c9771f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -324,7 +324,7 @@ def test_simple_pipeline(prompt_model):
pipe = Pipeline()
pipe.add_node(component=node, name="prompt_node", inputs=["Query"])
result = pipe.run(query="not relevant", documents=[Document("Berlin is an amazing city.")])
assert result["out"][0].casefold() == "positive"
assert "positive" in result["out"][0].casefold()
@pytest.mark.integration