public void TestSerialization()
{
Random r = new Random();
TDigest digestA = new TDigest();
for (int i = 0; i < 10000; i++)
{
var n = (r.Next() % 50) + (r.Next() % 50);
digestA.Add(n);
}
byte[] s = digestA.Serialize();
TDigest digestB = new TDigest(s);
var a = digestA.GetDistribution();
var b = digestB.GetDistribution();
for (int i = 0; i < a.Length; i++)
{
var ce = a[i].Count == b[i].Count;
var me = a[i].Value == b[i].Value;
Assert.IsTrue(ce && me, "Centroid means or counts are not equal after serialization");
}
Assert.AreEqual(digestA.Average, digestB.Average, "Averages are not equal after serialization");
Assert.AreEqual(digestA.Count, digestB.Count, "Counts are not equal after serialization");
Assert.AreEqual(digestA.CentroidCount, digestB.CentroidCount, "Centroid Counts are not equal after serialization");
Assert.AreEqual(digestA.CompressionConstant, digestB.CompressionConstant, "Compression Constants are not equal after serialization");
Assert.AreEqual(digestA.Accuracy, digestB.Accuracy, "Accuracies are not equal after serialization");
var differences = Enumerable.Range(1, 999)
.Select(n => n / 1000.0)
.Where(q => digestA.Quantile(q) != digestB.Quantile(q))
.Select(q => new { q, A = digestA.Quantile(q), B = digestB.Quantile(q) })
.ToList();
var areEqual = !differences.Any();
Assert.IsTrue(areEqual, "Serialized TDigest is not the same as original");
}
