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

Improved test_add_route

parent e43b9a1f
No related branches found
No related tags found
No related merge requests found
...@@ -155,11 +155,15 @@ class TestRouteLayer: ...@@ -155,11 +155,15 @@ 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"])
assert route_layer.routes == []
route_layer.add(route=route1) route_layer.add(route=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 assert route_layer.index.shape[0] == 2
route_layer.add(route=route2) route_layer.add(route=route2)
assert route_layer.routes == [route1, route2]
assert route_layer.index.shape[0] == 4 assert route_layer.index.shape[0] == 4
del route_layer del route_layer
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment