From 34135011c8770ac82b3a7c635f00e00f615e9934 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 7 Jan 2024 16:30:14 +0100
Subject: [PATCH] add test for encoder is None
---
tests/unit/test_layer.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 286e7ace..92577f34 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -119,6 +119,10 @@ class TestRouteLayer:
route_layer_openai = RouteLayer(encoder=openai_encoder)
assert route_layer_openai.score_threshold == 0.82
+ def test_initialization_no_encoder(self, openai_encoder):
+ route_layer_none = RouteLayer(encoder=None)
+ assert route_layer_none.score_threshold == openai_encoder.score_threshold
+
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
--
GitLab