yzyCli/u3d/Assets/LuaFramework/Editor/Treemap/Tests/TreemapTests.cs

using System;
using NUnit.Framework;
using UnityEngine;
using System.Collections.Generic;
using System.Collections;
using System.Linq;

namespace Treemap
{
	[TestFixture]
	public class TreemapTests
	{
		static readonly Rect s_Rect = new UnityEngine.Rect (0, 0, 100, 100);

		[Test]
		public void Empty()
		{
			Assert.Throws<ArgumentException>(() => Utility.GetTreemapRects (new float[0], s_Rect));
		}

		[Test]
		public void One()
		{
			var result = Utility.GetTreemapRects (new [] { 1.0f }, s_Rect);
			Assert.AreEqual (1, result.Length);
			Assert.AreEqual (s_Rect, result[0]);
		}

		[Test]
		public void TwoEquallySized()
		{
			var result = Utility.GetTreemapRects (new [] { 1.0f, 1.0f }, s_Rect);
			Assert.AreEqual (2, result.Length);

			Assert.AreEqual (new Rect (0, 0, 100, 50), result [0]);
			Assert.AreEqual (new Rect (0, 50, 100, 50), result [1]);
		}

		[Test]
		public void Zero()
		{
			Assert.Throws<ArgumentException>(() => Utility.GetTreemapRects (new [] { 0f }, s_Rect));
		}

		[Test]
		public void NegativeNumber()
		{
			Assert.Throws<ArgumentException>(() => Utility.GetTreemapRects (new [] { -3f }, s_Rect));
		}

		[Test, TestCaseSource("InterestingDataSets")]
		public void ResultRectsHaveCorrectArea(float[] floats)
		{
			var results = Utility.GetTreemapRects (floats, s_Rect);
			Assert.AreEqual (floats.Length, results.Length); 

			var sum = floats.Sum ();
			var totalSize = s_Rect.width * s_Rect.height;

			for(int i =0; i!= floats.Length; i++)
			{
				var f = floats [i];
				var r = results [i];

				var size = r.width * r.height;
				var ratio = size / totalSize;

				UnityEngine.Assertions.Assert.AreApproximatelyEqual (f / sum, ratio);
			}
		}

		[Test, TestCaseSource("InterestingDataSets")]
		public void ResultRectsDoNotOverlap(float[] floats)
		{
			var results = Utility.GetTreemapRects (floats, s_Rect);
			Assert.AreEqual (floats.Length, results.Length); 

			for(int i1 =0; i1!= floats.Length; i1++)
			{
				var rect1 = results [i1];
				for (int i2 = i1 + 1; i2 != floats.Length; i2++) {
					var rect2 = results [i2];

					float overlapSize = SizeOfOverlappingRectOf(rect1, rect2);

					const float tolerance = 0.000001f;
					if (overlapSize > tolerance * rect1.width * rect1.height || overlapSize > tolerance * rect2.width * rect2.height)
						throw new AssertionException ("too big overlap. overlapSize: " + overlapSize + " rect1:" + rect1 + " rect2: " + rect2);
				}
			}
		}

		private static Rect RectIntersect(Rect a, Rect b)
		{
			float xMin = Mathf.Max(a.x, b.x);
			float xMax = Mathf.Min(a.x + a.width, b.x + b.width);
			float yMin = Mathf.Max(a.y, b.y);
			float yMax = Mathf.Min(a.y + a.height, b.y + b.height);
			if (xMax >= xMin && yMax >= yMin)
				return new Rect(xMin, yMin, xMax - xMin, yMax - yMin);
			return new Rect(0f, 0f, 0f, 0f);
		}

		static float SizeOfOverlappingRectOf(Rect rect1, Rect rect2)
		{
			var result = RectIntersect (rect1, rect2);
			return result.width * result.height;
		}

		static IEnumerable InterestingDataSets()
		{
			yield return new TestCaseData (Enumerable.Range (100, 10).Select (i => (float)i).ToArray ()).SetName ("1000 floats counting up");
			yield return new TestCaseData (Enumerable.Repeat (10, 100).Select (i => (float)i).ToArray ()).SetName ("100 identical floats");
		}
	}
}