From 4ff71a1b19f5f05b703ae6e2fa2a96c9d47d31f9 Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Tue, 12 Mar 2024 01:10:50 +0400
Subject: [PATCH] Removed defunct PyTests.
---
tests/unit/test_layer.py | 35 -----------------------------------
1 file changed, 35 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index d7945450..80725108 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -523,38 +523,3 @@ class TestLayerConfig:
layer_config = LayerConfig(routes=[route])
layer_config.remove("test")
assert layer_config.routes == []
-
- def test_check_threshold_with_none_route(self, openai_encoder):
- route_layer = RouteLayer(encoder=openai_encoder)
- result = route_layer._check_threshold(scores=[0.5, 0.6], route=None)
- assert (
- not result
- ), "Expected _check_threshold to return False when route is None."
-
- def test_simulate_static_route_selection_returns_route_choice(
- self, openai_encoder, routes
- ):
- route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
- # Manually set the index to simulate a scenario where a route passes the threshold
- route_layer.index = LocalIndex()
- # Assuming routes[0] is the route we want to simulate as the top route
- route_layer.index.add(
- embeddings=[[0.1, 0.2, 0.3]],
- routes=[routes[0].name],
- utterances=[routes[0].utterances[0]],
- )
- # Adjust the score_threshold to ensure the route passes the threshold
- route_layer.score_threshold = 0
-
- # Simulate a vector that would match the route we added to the index
- vector = [0.1, 0.2, 0.3]
- route_choice = route_layer._simulate_static_route_selection(vector=vector)
-
- assert (
- route_choice.name == routes[0].name
- ), "Expected the RouteChoice name to match the simulated top route name."
- assert route_choice.function_call is None, "Expected function_call to be None."
- assert (
- route_choice.similarity_score is None
- ), "Expected similarity_score to be None."
- assert route_choice.trigger is None, "Expected trigger to be None."
--
GitLab