Update tests

Author: eliascarv

test/runtests.jl

@@ -40,6 +40,42 @@ using Test
     @test tree[1] == 11
     @test tree[3] == 33
 
+    # right rotate
+    tree = AVLTree{Int,Int}()
+    tree[3] = 30
+    tree[2] = 20
+    tree[1] = 10
+    @test tree.root.key == 2
+    @test tree.root.left.key == 1
+    @test tree.root.right.key == 3
+
+    # left rotate
+    tree = AVLTree{Int,Int}()
+    tree[1] = 10
+    tree[2] = 20
+    tree[3] = 30
+    @test tree.root.key == 2
+    @test tree.root.left.key == 1
+    @test tree.root.right.key == 3
+
+    # left-right rotate
+    tree = AVLTree{Int,Int}()
+    tree[3] = 30
+    tree[1] = 10
+    tree[2] = 20
+    @test tree.root.key == 2
+    @test tree.root.left.key == 1
+    @test tree.root.right.key == 3
+
+    # right-left rotate
+    tree = AVLTree{Int,Int}()
+    tree[1] = 10
+    tree[3] = 30
+    tree[2] = 20
+    @test tree.root.key == 2
+    @test tree.root.left.key == 1
+    @test tree.root.right.key == 3
+
     # delete
     tree = AVLTree{Int,Int}()
     tree[2] = 20
@@ -56,14 +92,24 @@ using Test
     delete!(tree, 2)
     @test isnothing(tree.root)
 
-    # tree that accept any types
+    # tree that accepts any types
     tree = AVLTree()
     tree[2] = 'A'
     tree[1] = 1.1
     tree[3] = "test"
     @test tree[2] == 'A'
     @test tree[1] == 1.1
     @test tree[3] == "test"
+    delete!(tree, 3)
+    @test !isnothing(tree.root)
+    @test !isnothing(tree.root.left)
+    @test isnothing(tree.root.right)
+    delete!(tree, 1)
+    @test !isnothing(tree.root)
+    @test isnothing(tree.root.left)
+    @test isnothing(tree.root.right)
+    delete!(tree, 2)
+    @test isnothing(tree.root)
 
     # AbstractTrees interface
     tree = AVLTree{Int,Int}()
@@ -86,42 +132,6 @@ using Test
     @test NodeType(root) === HasNodeType()
     @test nodetype(root) === typeof(root)
 
-    # left rotate
-    tree = AVLTree{Int,Int}()
-    tree[1] = 10
-    tree[2] = 20
-    tree[3] = 30
-    @test tree.root.key == 2
-    @test tree.root.left.key == 1
-    @test tree.root.right.key == 3
-
-    # right rotate
-    tree = AVLTree{Int,Int}()
-    tree[3] = 30
-    tree[2] = 20
-    tree[1] = 10
-    @test tree.root.key == 2
-    @test tree.root.left.key == 1
-    @test tree.root.right.key == 3
-
-    # left-right rotate
-    tree = AVLTree{Int,Int}()
-    tree[3] = 30
-    tree[1] = 10
-    tree[2] = 20
-    @test tree.root.key == 2
-    @test tree.root.left.key == 1
-    @test tree.root.right.key == 3
-
-    # right-left rotate
-    tree = AVLTree{Int,Int}()
-    tree[1] = 10
-    tree[3] = 30
-    tree[2] = 20
-    @test tree.root.key == 2
-    @test tree.root.left.key == 1
-    @test tree.root.right.key == 3
-
     # show
     tree = AVLTree{Int,Int}()
     @test sprint(show, MIME("text/plain"), tree) == "AVLTree()"