public void AddRoot()
{
TreeAlgorithm tree = new TreeAlgorithm();
tree.Add(56);
var expectedColor = Color.B;
var resultColor = tree.FindKey(56).Colour;
Assert.AreEqual(expectedColor, resultColor);
}
public void DeleteWithRedPrevTest()
{
TreeAlgorithm tree = new TreeAlgorithm();
int[] values = { 56, 12, 65, 16, 30 };
TreeComplite(tree, values);
tree.Delete1(56);
var result = tree.Root;
Assert.AreEqual(30, result.Value);
Assert.AreEqual(16, result.Left.Value);
Assert.IsNull(result.Left.Right);//старое местоположение узла 30 теперь должно быть пустым
Assert.IsNull(tree.FindKey(56));
}
public void DeleteWithRedPrevChildTest()
{
TreeAlgorithm tree = new TreeAlgorithm();
int[] values = { 56, 12, 65, 16, 30, 20 };
TreeComplite(tree, values);
tree.Delete1(56);
var result = tree.Root;
Assert.AreEqual(30, result.Value);
Assert.AreEqual(16, result.Left.Value);
Assert.AreEqual(Color.R, result.Left.Colour); //цвет узла 16(родителя) lдолжен поменяться на красный
Assert.AreEqual(20, result.Left.Right.Value); //на месте узла 30 теперь должен быть его потомок - узел 20
Assert.IsNull(tree.FindKey(56));
}
public void DeleteWithoutChildPrevTest()
{
TreeAlgorithm tree = new TreeAlgorithm();
int[] values = { 56, 12, 65, 16, 30, 20 };
TreeComplite(tree, values);
tree.Delete1(16);
var result = tree.Root;
Assert.AreEqual(56, result.Value);
Assert.AreEqual(20, result.Left.Value); //после поворота должен стать потомком корневого узла
Assert.AreEqual(Color.R, result.Left.Colour);
Assert.AreEqual(12, result.Left.Left.Value); // узел 12 должен стать потомком узла 20
Assert.AreEqual(Color.B, result.Left.Left.Colour);
Assert.AreEqual(30, result.Left.Right.Value); // узел 30 должен стать потомком узла 20
Assert.IsNull(tree.FindKey(16));
}
