Skip to content
Snippets Groups Projects
Unverified Commit e9660ee0 authored by James Briggs's avatar James Briggs
Browse files

added dynamic route init test

parent 3c3dce4c
No related branches found
No related tags found
No related merge requests found
...@@ -91,6 +91,13 @@ def routes(): ...@@ -91,6 +91,13 @@ def routes():
Route(name="Route 2", utterances=["Goodbye", "Bye", "Au revoir"]), 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: class TestRouteLayer:
def test_initialization(self, openai_encoder, routes): def test_initialization(self, openai_encoder, routes):
...@@ -106,7 +113,12 @@ class TestRouteLayer: ...@@ -106,7 +113,12 @@ class TestRouteLayer:
def test_initialization_different_encoders(self, cohere_encoder, openai_encoder): def test_initialization_different_encoders(self, cohere_encoder, openai_encoder):
route_layer_cohere = RouteLayer(encoder=cohere_encoder) route_layer_cohere = RouteLayer(encoder=cohere_encoder)
assert route_layer_cohere.score_threshold == 0.3 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) route_layer_openai = RouteLayer(encoder=openai_encoder)
assert route_layer_openai.score_threshold == 0.82 assert route_layer_openai.score_threshold == 0.82
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment