Skip to content
Snippets Groups Projects
Unverified Commit d3b5d8db authored by Siraj R Aizlewood's avatar Siraj R Aizlewood
Browse files

More PyTest Fixes

Using index.describe instead of shape[].
parent 31689076
Branches
Tags
No related merge requests found
...@@ -155,17 +155,20 @@ class TestRouteLayer: ...@@ -155,17 +155,20 @@ class TestRouteLayer:
route1 = Route(name="Route 1", utterances=["Yes", "No"]) route1 = Route(name="Route 1", utterances=["Yes", "No"])
route2 = Route(name="Route 2", utterances=["Maybe", "Sure"]) route2 = Route(name="Route 2", utterances=["Maybe", "Sure"])
# Initially, the routes list should be empty
assert route_layer.routes == [] assert route_layer.routes == []
# Add route1 and check
route_layer.add(route=route1) route_layer.add(route=route1)
assert route_layer.routes == [route1] assert route_layer.routes == [route1]
assert route_layer.index is not None assert route_layer.index is not None
assert route_layer.index.shape[0] == 2 # Use the describe method to get the number of vectors
assert route_layer.index.describe()['vectors'] == 2
# Add route2 and check
route_layer.add(route=route2) route_layer.add(route=route2)
assert route_layer.routes == [route1, route2] assert route_layer.routes == [route1, route2]
assert route_layer.index.shape[0] == 4 assert route_layer.index.describe()['vectors'] == 4
del route_layer
def test_list_route_names(self, openai_encoder, routes): def test_list_route_names(self, openai_encoder, routes):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes) route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
...@@ -203,7 +206,7 @@ class TestRouteLayer: ...@@ -203,7 +206,7 @@ class TestRouteLayer:
route_layer = RouteLayer(encoder=openai_encoder) route_layer = RouteLayer(encoder=openai_encoder)
route_layer._add_routes(routes=routes) route_layer._add_routes(routes=routes)
assert route_layer.index is not None assert route_layer.index is not None
assert route_layer.index.shape[0] == 5 assert route_layer.index.describe()['vectors'] == 5
def test_query_and_classification(self, openai_encoder, routes): def test_query_and_classification(self, openai_encoder, routes):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes) route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment