Fix DFS test.

PiperOrigin-RevId: 575241826
google · Oct 20, 2023 · 9e37edd · 9e37edd
1 parent 490e713
commit 9e37edd
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/java/tests/com/google/template/soy/internal/util/TreeStreamsTest.java b/java/tests/com/google/template/soy/internal/util/TreeStreamsTest.java
@@ -39,7 +39,8 @@ public void testBreadthFirst() {
                     .put(2, list(4, 5))
                     .put(3, list(6, 7))
                     .buildOrThrow()))
-        .containsExactly(1, 2, 3, 4, 5, 6, 7);
+        .containsExactly(1, 2, 3, 4, 5, 6, 7)
+        .inOrder();
   }
 
   @Test
@@ -52,7 +53,8 @@ public void testDepthFirst() {
                     .put(2, list(4, 5))
                     .put(3, list(6, 7))
                     .buildOrThrow()))
-        .containsExactly(1, 2, 4, 5, 3, 6, 7);
+        .containsExactly(1, 2, 4, 5, 3, 6, 7)
+        .inOrder();
   }
 
   private static <T> ImmutableList<T> list(T... items) {
@@ -65,7 +67,7 @@ private static <T> ImmutableList<T> bfs(T root, Map<T, Iterable<T>> data) {
   }
 
   private static <T> ImmutableList<T> dfs(T root, Map<T, Iterable<T>> data) {
-    return TreeStreams.breadthFirst(root, k -> data.getOrDefault(k, ImmutableList.of()))
+    return TreeStreams.depthFirst(root, k -> data.getOrDefault(k, ImmutableList.of()))
         .collect(ImmutableList.toImmutableList());
   }
 }