From 7862d39db4cab13e2cb49eea7387fa5f537432f8 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 4 Jan 2025 13:57:40 +0400
Subject: [PATCH] fix: missing assertion logic
---
tests/unit/test_router.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index cd4d1420..cfdc1f6b 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -295,7 +295,13 @@ class TestIndexEncoders:
encoder = encoder_cls()
index = init_index(index_cls, index_name=encoder.__class__.__name__)
route_layer = router_cls(encoder=encoder, index=index)
- assert route_layer.score_threshold == encoder.score_threshold
+ if isinstance(route_layer, HybridRouter):
+ assert (
+ route_layer.score_threshold
+ == encoder.score_threshold * route_layer.alpha
+ )
+ else:
+ assert route_layer.score_threshold == encoder.score_threshold
def test_initialization_no_encoder(self, index_cls, encoder_cls, router_cls):
os.environ["OPENAI_API_KEY"] = "test_api_key"
--
GitLab