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

Linting.

parent d3b5d8db
No related branches found
No related tags found
No related merge requests found
...@@ -163,12 +163,12 @@ class TestRouteLayer: ...@@ -163,12 +163,12 @@ class TestRouteLayer:
assert route_layer.routes == [route1] assert route_layer.routes == [route1]
assert route_layer.index is not None assert route_layer.index is not None
# Use the describe method to get the number of vectors # Use the describe method to get the number of vectors
assert route_layer.index.describe()['vectors'] == 2 assert route_layer.index.describe()["vectors"] == 2
# Add route2 and check # 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.describe()['vectors'] == 4 assert route_layer.index.describe()["vectors"] == 4
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)
...@@ -206,7 +206,7 @@ class TestRouteLayer: ...@@ -206,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.describe()['vectors'] == 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.
Finish editing this message first!
Please register or to comment