最近两天写单元测试，碰到需要验证一个树是否是期望的，但是树这个结构要验证起来还真是有点烦。。。
我的树大概是这样的：
1 class TreeNode<T> 2 { 3 private static readonly TreeNode<T>[] Empty = new TreeNode<T>[0]; 4 public TreeNode() : this(default(T), Empty) { } 5 public TreeNode(T value, IReadOnlyList<TreeNode<T>> children) 6 { 7 Value = value; 8 Children = children; 9 } 10 public T Value { get; set; } 11 public IReadOnlyList<TreeNode<T>> Children { get; set; } 12 }
View Code
为了判等，这个树实现了IEquatable<TreeNode<T>>接口：
1 public override bool Equals(object obj) 2 { 3 var other = obj as TreeNode<T>; 4 if (other == null) 5 { 6 return false; 7 } 8 return Equals(other); 9 } 10 11 public bool Equals(TreeNode<T> other) 12 { 13 return object.Equals(Value, other.Value) && Children.SequenceEqual(other.Children); 14 } 15 16 public override int GetHashCode() 17 { 18 return Value.GetHashCode() ^ Children.Aggregate(Children.Count, (seed, x) => seed ^ x.GetHashCode()); 19 }
View Code
看着还不错，不过由于T实际是个复杂类型，每次重写Equals也是个不轻松的事情，而且还要把整个期望的树给构造出来，呵呵，还是烦啊。。。
但是，如果只需要判定个把简单属性，事情就方便了许多，所以，TreeNode需要一个方法来转换T的类型：
1 public TreeNode<TResult> Select<TResult>(Func<T, TResult> selector) 2 { 3 return new TreeNode<TResult>(selector(Value), (from c in Children select c.Select(selector)).ToList()); 4 }
View Code
看起来不错，这样就可以有这样的code来玩转tree了：
1 TreeNode<int> intTree = ... 2 TreeNode<string> stringTree = intTree.Select(i => i.ToString());
View Code
等等，我们可以把这代码写的跟linq：
1 TreeNode<int> intTree = ... 2 TreeNode<string> stringTree = from i in intTree 3 select i.ToString();
View Code
测试代码继续啪啦啪啦的写，唉，测试失败了，什么情况，仔细一看，哦，tree下面节点的顺序错了，而tree的equals方法要求顺序，但是这个测试刚好不要求顺序，于是我有了两个选择：
1. 改写equals方法 （不想折腾集合操作）
2. 让节点排序 （对测试用例而言，构建一个有顺序的树可是很简单的事情）
所以，我需要个OrderBy:
1 public TreeNode<T> OrderBy<TKey>(Func<T, TKey> keySelector) 2 { 3 return new TreeNode<T>(Value, (from c in Children 4 orderby keySelector(c.Value) 5 select c.OrderBy(keySelector)).ToList()); 6 }
View Code
这下就可以随便折腾这个树了：
1 TreeNode<int> intTree = ... 2 TreeNode<string> stringTree = from i in intTree 3 let m = i % 3 4 order by m 5 select i.ToString();
View Code
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。