Skip to content
Snippets Groups Projects
Commit a4504514 authored by Ismail Ashraq's avatar Ismail Ashraq
Browse files

update unit tests

parent 91da2413
Branches
Tags
No related merge requests found
......@@ -83,7 +83,7 @@ class TestAzureOpenAIEncoder:
)
with pytest.raises(ValueError) as e:
openai_encoder(["test document"])
assert "No embeddings returned. Error" in str(e.value)
assert "No embeddings returned." in str(e.value)
def test_openai_encoder_call_failure_non_openai_error(self, openai_encoder, mocker):
mocker.patch("os.getenv", return_value="fake-api-key")
......
......@@ -77,7 +77,7 @@ class TestOpenAIEncoder:
)
with pytest.raises(ValueError) as e:
openai_encoder(["test document"])
assert "No embeddings returned. Error" in str(e.value)
assert "No embeddings returned." in str(e.value)
def test_openai_encoder_call_failure_non_openai_error(self, openai_encoder, mocker):
mocker.patch("os.getenv", return_value="fake-api-key")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment