Issue #3 more tests

kinow · Dec 26, 2020 · 12f4c30 · 12f4c30
1 parent b0a8291
commit 12f4c30
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_node_iterators.py b/tests/test_node_iterators.py
@@ -46,9 +46,14 @@ def test_tasks_iterator(sample_graph):
 
     iterator = TasksIterator(graph)
 
-    expected = []
+    expected = [['a.0'], ['b.0', 'e.0', 'a.1'], ['c.0', 'b.1'], ['d.0', 'c.1'], ['d.1']]
     iterated = []
 
+    for index, node in enumerate(iterator):
+        iterated.append(node)
+        if index == 7:
+            break
+
     # TODO: use iterator
 
     assert iterated == expected