diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index 69a648339d9a3c5d162275547a74bed66d97f236..9669d594a0e560bea125c75d8cc5cd46caaac7df 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -170,7 +170,9 @@ class TestRouteLayer: def test_list_route_names(self, openai_encoder, routes): route_layer = RouteLayer(encoder=openai_encoder, routes=routes) route_names = route_layer.list_route_names() - assert set(route_names) == {route.name for route in routes}, "The list of route names should match the names of the routes added." + assert set(route_names) == { + route.name for route in routes + }, "The list of route names should match the names of the routes added." def test_remove_route(self, openai_encoder, routes): route_layer = RouteLayer(encoder=openai_encoder, routes=routes) @@ -178,12 +180,18 @@ class TestRouteLayer: route_to_remove = routes[0].name route_layer.remove(route_to_remove) # Ensure the route is no longer in the route layer - assert route_to_remove not in route_layer.list_route_names(), "The route should be removed from the route layer." + assert ( + route_to_remove not in route_layer.list_route_names() + ), "The route should be removed from the route layer." # Ensure the route is no longer in the index or categories - assert route_to_remove not in route_layer.categories, "The route should be removed from the categories." + assert ( + route_to_remove not in route_layer.categories + ), "The route should be removed from the categories." # Ensure the route's utterances are no longer in the index for utterance in routes[0].utterances: - assert utterance not in route_layer.index, "The route's utterances should be removed from the index." + assert ( + utterance not in route_layer.index + ), "The route's utterances should be removed from the index." def test_remove_route_not_found(self, openai_encoder, routes): route_layer = RouteLayer(encoder=openai_encoder, routes=routes) @@ -191,7 +199,9 @@ class TestRouteLayer: non_existent_route = "non-existent-route" with pytest.raises(ValueError) as excinfo: route_layer.remove(non_existent_route) - assert str(excinfo.value) == f"Route `{non_existent_route}` not found", "Attempting to remove a non-existent route should raise a ValueError." + assert ( + str(excinfo.value) == f"Route `{non_existent_route}` not found" + ), "Attempting to remove a non-existent route should raise a ValueError." def test_add_multiple_routes(self, openai_encoder, routes): route_layer = RouteLayer(encoder=openai_encoder)