From e6f14149a2ab225fe7fd26296d9fff6cbd1e0f4f Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Mon, 30 Sep 2024 15:38:10 +0400
Subject: [PATCH] Linting.
---
tests/unit/test_layer.py | 22 +++++++++++++++-------
1 file changed, 15 insertions(+), 7 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 8b985c13..3bfcd485 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -818,30 +818,39 @@ class TestRouteLayer:
new_threshold = 0.8
route_layer.update(name=route_name, threshold=new_threshold)
updated_route = route_layer.get(route_name)
- assert updated_route.score_threshold == new_threshold, f"Expected threshold to be updated to {new_threshold}, but got {updated_route.score_threshold}"
+ assert (
+ updated_route.score_threshold == new_threshold
+ ), f"Expected threshold to be updated to {new_threshold}, but got {updated_route.score_threshold}"
def test_update_non_existent_route(self, openai_encoder, routes, index_cls):
index = init_index(index_cls)
route_layer = RouteLayer(encoder=openai_encoder, routes=routes, index=index)
non_existent_route = "Non-existent Route"
- with pytest.raises(ValueError, match=f"Route '{non_existent_route}' not found. Nothing updated."):
+ with pytest.raises(
+ ValueError,
+ match=f"Route '{non_existent_route}' not found. Nothing updated.",
+ ):
route_layer.update(name=non_existent_route, threshold=0.7)
def test_update_without_parameters(self, openai_encoder, routes, index_cls):
index = init_index(index_cls)
route_layer = RouteLayer(encoder=openai_encoder, routes=routes, index=index)
- with pytest.raises(ValueError, match="At least one of 'threshold' or 'utterances' must be provided."):
+ with pytest.raises(
+ ValueError,
+ match="At least one of 'threshold' or 'utterances' must be provided.",
+ ):
route_layer.update(name="Route 1")
def test_update_utterances_not_implemented(self, openai_encoder, routes, index_cls):
index = init_index(index_cls)
route_layer = RouteLayer(encoder=openai_encoder, routes=routes, index=index)
- with pytest.raises(NotImplementedError, match="The update method cannot be used for updating utterances yet."):
+ with pytest.raises(
+ NotImplementedError,
+ match="The update method cannot be used for updating utterances yet.",
+ ):
route_layer.update(name="Route 1", utterances=["New utterance"])
-
-
class TestLayerFit:
def test_eval(self, openai_encoder, routes, test_data):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
@@ -978,4 +987,3 @@ class TestLayerConfig:
elif agg == "max":
assert classification == "Route 3"
assert score == [0.1, 1.0]
-
--
GitLab