def test_search_node(self):
node = BSTNode(STestClass("A", "a"))
node.add(STestClass("G", "g"))
node.add(STestClass("H", "h"))
node.add(STestClass("B", "b"))
node.add(STestClass("Z", "z"))
search_node = node.search_node(STestClass("G", "g"))
assert search_node._element.full_str() == "G: g"
def test_search(self):
node = BSTNode(STestClass("A", "a"))
node.add(STestClass("G", "g"))
node.add(STestClass("H", "h"))
node.add(STestClass("B", "b"))
node.add(STestClass("Z", "z"))
search_item = node.search_node(STestClass("C", "c"))
assert search_item is None
search_item = node.search(STestClass("H", "h"))
assert search_item.full_str() == "H: h"
