public void TestSmartArbiter()
{
var fu = new FindUnion <int>(TakeEvenArbiter);
fu.PushCheckpoint();
fu.PushCheckpoint();
fu.PushCheckpoint();
fu.Union(0, 1);
Assert.AreEqual(0, fu.GetRepresentant(0));
Assert.AreEqual(0, fu.GetRepresentant(1));
fu.PopCheckpoint();
fu.Union(1, 0);
Assert.AreEqual(0, fu.GetRepresentant(0));
Assert.AreEqual(0, fu.GetRepresentant(1));
fu.PopCheckpoint();
for (int i = 3; i < 10; i += 2)
{
fu.Union(1, i);
}
fu.Union(0, 1);
Assert.AreEqual(0, fu.GetRepresentant(0));
for (int i = 1; i < 10; i += 2)
{
Assert.AreEqual(0, fu.GetRepresentant(i));
}
fu.PopCheckpoint();
Assert.AreEqual(0, fu.GetRepresentant(0));
for (int i = 1; i < 10; i += 2)
{
Assert.AreEqual(i, fu.GetRepresentant(i));
}
}
public void TestCheckpointDoubleRestore()
{
var fu = new FindUnion <int>(FixedArbiter <int>);
fu.PushCheckpoint();
for (int i = 2; i < 10; i++)
{
fu.Union(i % 2, i);
}
for (int i = 0; i < 10; i += 2)
{
Assert.AreEqual(0, fu.GetRepresentant(i));
Assert.AreEqual(1, fu.GetRepresentant(i + 1));
}
fu.PushCheckpoint();
fu.Union(0, 1);
for (int i = 0; i < 10; i++)
{
Assert.AreEqual(0, fu.GetRepresentant(i));
}
fu.PopCheckpoint();
for (int i = 0; i < 10; i += 2)
{
Assert.AreEqual(0, fu.GetRepresentant(i));
Assert.AreEqual(1, fu.GetRepresentant(i + 1));
}
fu.PopCheckpoint();
for (int i = 0; i < 10; i++)
{
Assert.AreEqual(i, fu.GetRepresentant(i));
}
}
