From a45045146f2025c1affccb545d40072cb7e785fe Mon Sep 17 00:00:00 2001
From: Ismail Ashraq <ismailashraq@Ismails-MacBook-Pro.local>
Date: Sun, 18 Aug 2024 16:39:57 +0500
Subject: [PATCH] update unit tests
---
tests/unit/encoders/test_azure.py | 2 +-
tests/unit/encoders/test_openai.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/unit/encoders/test_azure.py b/tests/unit/encoders/test_azure.py
index 93dffb89..3969543e 100644
--- a/tests/unit/encoders/test_azure.py
+++ b/tests/unit/encoders/test_azure.py
@@ -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")
diff --git a/tests/unit/encoders/test_openai.py b/tests/unit/encoders/test_openai.py
index 508e9e9e..56ef2b38 100644
--- a/tests/unit/encoders/test_openai.py
+++ b/tests/unit/encoders/test_openai.py
@@ -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")
--
GitLab