diff --git a/tests/unit/utils/test_config.py b/tests/unit/utils/test_config.py index b0386f18..51b5be48 100644 --- a/tests/unit/utils/test_config.py +++ b/tests/unit/utils/test_config.py @@ -411,7 +411,8 @@ def test_invalid_config_for_memory_cache(): password_path: pwd123 """, InvalidConfigurationError, - "memory conversation cache type is specified, but memory configuration is missing", + "memory conversation cache type is specified," + " but memory configuration is missing", ) check_expected_exception( @@ -459,7 +460,8 @@ def test_invalid_config_for_redis_cache(): max_entries: 1000 """, InvalidConfigurationError, - "redis conversation cache type is specified, but redis configuration is missing", + "redis conversation cache type is specified," + " but redis configuration is missing", ) check_expected_exception( diff --git a/tests/unit/utils/test_errors_parsing.py b/tests/unit/utils/test_errors_parsing.py index 0f7f1d90..d712700b 100644 --- a/tests/unit/utils/test_errors_parsing.py +++ b/tests/unit/utils/test_errors_parsing.py @@ -1,4 +1,4 @@ -"""Unit Test for the Utils class.""" +"""Unit Test for the errors parsing class.""" from genai.exceptions import ApiResponseException from httpx import Response diff --git a/tests/unit/utils/test_pyroscope.py b/tests/unit/utils/test_pyroscope.py index 68a97acd..c3c5e76d 100644 --- a/tests/unit/utils/test_pyroscope.py +++ b/tests/unit/utils/test_pyroscope.py @@ -6,7 +6,9 @@ import pytest from requests.exceptions import RequestException -from ols.utils.pyroscope import start_with_pyroscope_enabled +from ols.utils.pyroscope import ( + start_with_pyroscope_enabled, +) @pytest.fixture diff --git a/tests/unit/utils/test_redactor.py b/tests/unit/utils/test_redactor.py index 98545c8d..9564eb04 100644 --- a/tests/unit/utils/test_redactor.py +++ b/tests/unit/utils/test_redactor.py @@ -30,7 +30,10 @@ def test_redact_question_image_ip(self): "write a deployment yaml for the mongodb image with nodeip as 1.123.0.99" ) redacted_question = self.query_filter.redact("test_id", query) - expected_output = "write a deployment yaml for the mongodb REDACTED_image with nodeip as REDACTED_IP" # noqa: E501 + expected_output = ( + "write a deployment yaml for the mongodb REDACTED_image with nodeip " + "as REDACTED_IP" + ) assert redacted_question == expected_output def test_redact_question_mongopart_url_phone(self):