added tests

Author: Kishan-Ved

pydatastructs/trees/_backend/cpp/AVLTree.hpp

@@ -229,14 +229,20 @@ static PyObject* AVLTree_insert(AVLTree* self, PyObject *args) {
         return NULL;
     }
     SelfBalancingBinaryTree_insert(self->sbbt, Py_BuildValue("(OO)", key, data));
-    long s = self->sbbt->bst->binary_tree->size - 1;
+    BinaryTree* bt = self->sbbt->bst->binary_tree;
+    long s = bt->size - 1;
     AVLTree__balance_insertion(self, Py_BuildValue("(OO)", PyLong_FromLong(s), reinterpret_cast<TreeNode*>(bt->tree->_one_dimensional_array->_data[s])->parent));
 
     Py_RETURN_NONE;
 }
 
 static struct PyMethodDef AVLTree_PyMethodDef[] = {
     {"search", (PyCFunction) AVLTree_search, METH_VARARGS | METH_KEYWORDS, NULL},
+    {"insert", (PyCFunction) AVLTree_insert, METH_VARARGS, NULL},
+    {"_left_right_rotate", (PyCFunction) AVLTree__left_right_rotate, METH_VARARGS, NULL},
+    {"_right_left_rotate", (PyCFunction) AVLTree__right_left_rotate, METH_VARARGS, NULL},
+    {"_left_rotate", (PyCFunction) AVLTree__left_rotate, METH_VARARGS, NULL},
+    {"_right_rotate", (PyCFunction) AVLTree__right_rotate, METH_VARARGS, NULL},
     {NULL}
 };
 

pydatastructs/trees/_backend/cpp/BinaryTreeTraversal.hpp

@@ -16,6 +16,7 @@
 #include "SelfBalancingBinaryTree.hpp"
 #include "RedBlackTree.hpp"
 #include "SplayTree.hpp"
+#include "AVLTree.hpp"
 
 typedef struct {
     PyObject_HEAD
@@ -44,10 +45,16 @@ static PyObject* BinaryTreeTraversal___new__(PyTypeObject* type, PyObject *args,
     if (PyType_Ready(&SplayTreeType) < 0) { // This has to be present to finalize a type object. This should be called on all type objects to finish their initialization.
         return NULL;
     }
+    if (PyType_Ready(&AVLTreeType) < 0) { // This has to be present to finalize a type object. This should be called on all type objects to finish their initialization.
+        return NULL;
+    }
 
     if (PyObject_IsInstance(tree, (PyObject *)&SplayTreeType)) {
         self->tree = reinterpret_cast<SplayTree*>(tree)->sbbt->bst->binary_tree;
     }
+    else if (PyObject_IsInstance(tree, (PyObject *)&AVLTreeType)) {
+        self->tree = reinterpret_cast<AVLTree*>(tree)->sbbt->bst->binary_tree;
+    }
     else if (PyObject_IsInstance(tree, (PyObject *)&RedBlackTreeType)) {
         self->tree = reinterpret_cast<RedBlackTree*>(tree)->sbbt->bst->binary_tree;
     }