Skip to content

Commit b57d4f7

Browse files
committed
Adds more checkInvariants() calls.
1 parent 7dd2de0 commit b57d4f7

File tree

3 files changed

+6
-1
lines changed

3 files changed

+6
-1
lines changed

src/main/java/org/javimmutable/collections/hash/hamt/HamtBranchNode.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,9 @@ public void checkInvariants()
340340
throw new IllegalStateException();
341341
}
342342
}
343+
for (HamtNode<T, K, V> child : children) {
344+
child.checkInvariants();
345+
}
343346
}
344347

345348
private Indexed<SplitableIterable<T>> indexedForIterator()

src/test/java/org/javimmutable/collections/hash/JImmutableHashMapTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,8 @@ public void testRandom1()
125125
case 4:
126126
key = random.nextInt(maxKey);
127127
assertEquals(expected.get(key), map.find(key).getValueOrNull());
128-
129128
assertEquals(expected.size(), map.size());
129+
map.checkInvariants();
130130
}
131131
}
132132

src/test/java/org/javimmutable/collections/hash/hamt/HamtBranchNodeTest.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -326,6 +326,7 @@ public void testRandom()
326326
node = node.assign(transforms, key, key, key, size);
327327
assertEquals(node.getValueOr(transforms, key, key, -1), node.find(transforms, key, key).getValueOr(-1));
328328
}
329+
node.checkInvariants();
329330
verifyIntContents(transforms, node, domain);
330331

331332
final MutableDelta zero = new MutableDelta();
@@ -336,6 +337,7 @@ public void testRandom()
336337
assertEquals(null, node.getValueOr(transforms, key, key, null));
337338
assertEquals(null, node.find(transforms, key, key).getValueOr(null));
338339
}
340+
node.checkInvariants();
339341
assertSame(HamtEmptyNode.of(), node);
340342
assertEquals(0, size.getValue());
341343
assertEquals(0, zero.getValue());

0 commit comments

Comments
 (0)