backend redirected to C++

Author: Kishan-Ved

pydatastructs/trees/binary_trees.py

@@ -1558,8 +1558,9 @@ def methods(cls):
     __slots__ = ['tree']
 
     def __new__(cls, tree, **kwargs):
-        raise_if_backend_is_not_python(
-            cls, kwargs.get('backend', Backend.PYTHON))
+        backend = kwargs.get('backend', Backend.PYTHON)
+        if backend == Backend.CPP:
+            return _trees.BinaryTreeTraversal(tree, **kwargs)
         if not isinstance(tree, BinaryTree):
             raise TypeError("%s is not a binary tree"%(tree))
         obj = object.__new__(cls)

pydatastructs/trees/tests/test_binary_trees.py

@@ -60,10 +60,9 @@ def test_cpp_BST2():
     "(5, 6, 6, 6), (None, 4, 4, None), (None, 7, 7, None), (8, 14, 14, None), "
     "(None, 13, 13, None)]")
 
-    ##### BinaryTreeTraversal _pre_order test #####
-    from pydatastructs.trees._backend.cpp import _trees
-    t = _trees.BinaryTreeTraversal(b)
-    p = t.depth_first_search(order="pre_order")
+    ##### BinaryTreeTraversal pre_order test #####
+    trav = BinaryTreeTraversal(b, backend=Backend.CPP)
+    p = trav.depth_first_search(order="pre_order")
     assert [node.key for node in p] == [8, 3, 1, 6, 4, 7, 10, 14, 13]
 
     ##### _simple_path() test #####