From c70d08616296b81ab3d8d4fad33bb746560bc30c Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 2 Dec 2023 12:48:19 -0800
Subject: [PATCH] update tests to use score_threshold
---
tests/unit/test_layer.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 96e06a08..0b9842fb 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -45,16 +45,16 @@ def decisions():
class TestDecisionLayer:
def test_initialization(self, openai_encoder, decisions):
decision_layer = DecisionLayer(encoder=openai_encoder, decisions=decisions)
- assert decision_layer.similarity_threshold == 0.82
+ assert decision_layer.score_threshold == 0.82
assert len(decision_layer.index) == 5
assert len(set(decision_layer.categories)) == 2
def test_initialization_different_encoders(self, cohere_encoder, openai_encoder):
decision_layer_cohere = DecisionLayer(encoder=cohere_encoder)
- assert decision_layer_cohere.similarity_threshold == 0.3
+ assert decision_layer_cohere.score_threshold == 0.3
decision_layer_openai = DecisionLayer(encoder=openai_encoder)
- assert decision_layer_openai.similarity_threshold == 0.82
+ assert decision_layer_openai.score_threshold == 0.82
def test_add_decision(self, openai_encoder):
decision_layer = DecisionLayer(encoder=openai_encoder)
@@ -107,9 +107,9 @@ class TestDecisionLayer:
assert not decision_layer._pass_threshold([], 0.5)
assert decision_layer._pass_threshold([0.6, 0.7], 0.5)
- def test_failover_similarity_threshold(self, base_encoder):
+ def test_failover_score_threshold(self, base_encoder):
decision_layer = DecisionLayer(encoder=base_encoder)
- assert decision_layer.similarity_threshold == 0.82
+ assert decision_layer.score_threshold == 0.82
# Add more tests for edge cases and error handling as needed.
--
GitLab