From e43b9a1fa444667167f5a626a3c68f81bcbecde6 Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Mon, 12 Feb 2024 13:18:31 +0400
Subject: [PATCH] Fixed PyTests
The RouteLayer.categories() attribute has been removed, removing parts that reference this attribute form the tests.
---
tests/unit/test_layer.py | 13 ++-----------
1 file changed, 2 insertions(+), 11 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 4e277e86..97353f97 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -156,15 +156,11 @@ class TestRouteLayer:
route2 = Route(name="Route 2", utterances=["Maybe", "Sure"])
route_layer.add(route=route1)
- assert route_layer.index is not None and route_layer.categories is not None
+ assert route_layer.index is not None
assert route_layer.index.shape[0] == 2
- assert len(set(route_layer.categories)) == 1
- assert set(route_layer.categories) == {"Route 1"}
route_layer.add(route=route2)
assert route_layer.index.shape[0] == 4
- assert len(set(route_layer.categories)) == 2
- assert set(route_layer.categories) == {"Route 1", "Route 2"}
del route_layer
def test_list_route_names(self, openai_encoder, routes):
@@ -183,10 +179,6 @@ class TestRouteLayer:
assert (
route_to_delete not in route_layer.list_route_names()
), "The route should be deleted from the route layer."
- # Ensure the route is no longer in the index or categories
- assert (
- route_to_delete not in route_layer.categories
- ), "The route should be deleted from the categories."
# Ensure the route's utterances are no longer in the index
for utterance in routes[0].utterances:
assert (
@@ -206,9 +198,8 @@ class TestRouteLayer:
def test_add_multiple_routes(self, openai_encoder, routes):
route_layer = RouteLayer(encoder=openai_encoder)
route_layer._add_routes(routes=routes)
- assert route_layer.index is not None and route_layer.categories is not None
+ assert route_layer.index is not None
assert route_layer.index.shape[0] == 5
- assert len(set(route_layer.categories)) == 2
def test_query_and_classification(self, openai_encoder, routes):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
--
GitLab