Vis*_*ota 2 java collections junit
我正试图回答Cracking the Coding Interview中的以下问题.下面的代码是GitHub项目的一部分,在这里.
给定一个二叉搜索树,设计一个算法,创建每个深度的所有节点的链表(即,如果你有一个深度为D的树,你将有D个链表).
作为一个优秀的小开发人员,我写了一个单元测试来检查这个.
@Test
public void testGetValuesAtEachLevel() {
Integer[] treeValues = {
1, 2, 3, 4, 5,
6, 7, 8, 9, 10,
11, 12, 13, 14, 15
};
tree = new GenericBinaryTree<>(treeValues);
Integer[][] expectedArray = {
{ 1 },
{ 2, 3 },
{ 4, 5, 6, 7 },
{ 8, 9, 10, 11, 12, 13, 14, 15 }
};
List<List<Node<Integer>>> expectedList = new ArrayList<>(4);
for (Integer[] level : expectedArray) {
List<Node<Integer>> list = new LinkedList<>();
for (Integer value : level) {
list.add(new Node<>(value));
}
expectedList.add(list);
}
assertEquals(expectedList, tree.getValuesAtEachLevel());
}
Run Code Online (Sandbox Code Playgroud)
这是代码.
List<List<Node<T>>> getValuesAtEachLevel() {
List<List<Node<T>>> results = new ArrayList<>();
List<Node<T>> firstLevel = new LinkedList<>();
firstLevel.add(getRoot());
results.add(firstLevel);
loadAtLevel(results, 1);
return results;
}
private void loadAtLevel(List<List<Node<T>>> list, int level) {
List<Node<T>> levelList = new LinkedList<Node<T>>();
for (Node<T> node : list.get(level - 1)) {
if (node.left() != null) levelList.add(node.left());
if (node.right() != null) levelList.add(node.right());
}
if (levelList.isEmpty()) return;
level++;
list.add(levelList);
loadAtLevel(list, level);
}
Run Code Online (Sandbox Code Playgroud)
想象一下,当单元测试因以下错误而失败时我会感到惊讶:
java.lang.AssertionError: expected:
java.util.ArrayList<[[1], [2, 3], [4, 5, 6, 7], [8, 9, 10, 11, 12, 13, 14, 15]]> but was:
java.util.ArrayList<[[1], [2, 3], [4, 5, 6, 7], [8, 9, 10, 11, 12, 13, 14, 15]]>
Run Code Online (Sandbox Code Playgroud)
字符串表示是相同的,所以我无法弄清楚这里发生了什么.事实上,如果我将断言更改为此,则测试通过:
assertEquals(expectedList.toString(), tree.getValuesAtEachLevel().toString());
Run Code Online (Sandbox Code Playgroud)
我意识到我即将学习有关接口和对象的非常有价值的东西.我在这做错了什么?
正如@Jon Skeet在评论中指出的那样,你需要覆盖equals和hashCode你的Node类,equals在它的实例上执行compare ().
一个样本(使用intelliJ默认自动生成)可能是 -
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Node<?> node = (Node<?>) o;
if (left != null ? !left.equals(node.left) : node.left != null) return false;
if (right != null ? !right.equals(node.right) : node.right != null) return false;
if (parent != null ? !parent.equals(node.parent) : node.parent != null) return false;
return value != null ? value.equals(node.value) : node.value == null;
}
@Override
public int hashCode() {
int result = left != null ? left.hashCode() : 0;
result = 31 * result + (right != null ? right.hashCode() : 0);
result = 31 * result + (parent != null ? parent.hashCode() : 0);
result = 31 * result + (value != null ? value.hashCode() : 0);
return result;
}
Run Code Online (Sandbox Code Playgroud)
另一方面为什么呢
assertEquals(expectedList.toString(),tree.getValuesAtEachLevel().toString());
Run Code Online (Sandbox Code Playgroud)
作品.
这是因为你在这里最终断言的是比较两个在这里String有@Overriden equals定义的实例
| 归档时间: |
|
| 查看次数: |
8496 次 |
| 最近记录: |