From 1f2b9f0efdf75262418bc59a1f1c449649b20463 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 15 Dec 2024 14:22:27 +0400
Subject: [PATCH] fix: modify test to catch error
---
tests/unit/test_router.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index c7dec080..be10e3b6 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -152,7 +152,7 @@ def routes_4():
@pytest.fixture
def route_single_utterance():
return [
- Route(name="Route 1", utterances=["Hello"]),
+ Route(name="Route 3", utterances=["Hello"]),
]
@@ -259,11 +259,12 @@ class TestSemanticRouter:
assert route_layer.score_threshold == openai_encoder.score_threshold
def test_add_single_utterance(
- self, route_single_utterance, openai_encoder, index_cls
+ self, routes, route_single_utterance, openai_encoder, index_cls
):
index = init_index(index_cls)
route_layer = SemanticRouter(
encoder=openai_encoder,
+ routes=routes,
index=index,
auto_sync="local",
)
--
GitLab