From e9660ee0388f0aa2798786c41dfb0c54d0c6f659 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 7 Jan 2024 15:48:50 +0100
Subject: [PATCH] added dynamic route init test
---
tests/unit/test_layer.py | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 32754997..1511a983 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -91,6 +91,13 @@ def routes():
Route(name="Route 2", utterances=["Goodbye", "Bye", "Au revoir"]),
]
+@pytest.fixture
+def dynamic_routes():
+ return [
+ Route(name="Route 1", utterances=["Hello", "Hi"], function_schema="test"),
+ Route(name="Route 2", utterances=["Goodbye", "Bye", "Au revoir"]),
+ ]
+
class TestRouteLayer:
def test_initialization(self, openai_encoder, routes):
@@ -106,7 +113,12 @@ class TestRouteLayer:
def test_initialization_different_encoders(self, cohere_encoder, openai_encoder):
route_layer_cohere = RouteLayer(encoder=cohere_encoder)
assert route_layer_cohere.score_threshold == 0.3
+ route_layer_openai = RouteLayer(encoder=openai_encoder)
+ assert route_layer_openai.score_threshold == 0.82
+ def test_initialization_dynamic_route(self, cohere_encoder, openai_encoder):
+ route_layer_cohere = RouteLayer(encoder=cohere_encoder)
+ assert route_layer_cohere.score_threshold == 0.3
route_layer_openai = RouteLayer(encoder=openai_encoder)
assert route_layer_openai.score_threshold == 0.82
--
GitLab