From 316890769f99dc2156fa8a461768c50ad809de91 Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Mon, 12 Feb 2024 13:41:12 +0400
Subject: [PATCH] Improved test_add_route
---
tests/unit/test_layer.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 97353f97..af025648 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -155,11 +155,15 @@ class TestRouteLayer:
route1 = Route(name="Route 1", utterances=["Yes", "No"])
route2 = Route(name="Route 2", utterances=["Maybe", "Sure"])
+ assert route_layer.routes == []
+
route_layer.add(route=route1)
+ assert route_layer.routes == [route1]
assert route_layer.index is not None
assert route_layer.index.shape[0] == 2
route_layer.add(route=route2)
+ assert route_layer.routes == [route1, route2]
assert route_layer.index.shape[0] == 4
del route_layer
--
GitLab