コード例 #1
0
def test_in_order():
    tree = BinarySearchTree()
    tree.add(10)
    tree.add(15)

    expected = [10]
    actual = tree.in_order()

    assert expected == actual
コード例 #2
0
def test_find_max_root_is_max():
  tree = BinarySearchTree()
  tree.root = Node(23)
  tree.root.left = Node(19)
  tree.root.right = Node(7)
  tree.root.left.left = Node(5)
  tree.root.left.right = Node(8)

  assert tree.find_max_val(tree.root) == 23
コード例 #3
0
def test_pre_order():
    tree = BinarySearchTree()
    tree.add(10)
    tree.add(5)
    tree.add(15)
    expected = [10,5,15]
    actual = tree.pre_order()
    assert expected == actual
コード例 #4
0
def test_three_members():
    tree = BinarySearchTree()
    tree.add(10)
    tree.add(5)
    tree.add(15)
    assert tree.root.value == 10
    assert tree.root.left.value == 5
    assert tree.root.right.value == 15
コード例 #5
0
def test_breadth_first_empty_list():

    tree = BinarySearchTree()

    assert tree.breadth_first() == 'tree is empty'
コード例 #6
0
def test_breadth_first():

    tree = BinarySearchTree()
    tree.add(2)
    tree.add(7)
    tree.add(5)
    tree.add(2)
    tree.add(6)
    tree.add(9)
    tree.add(5)
    tree.add(11)
    tree.add(4)

    assert tree.breadth_first() == [2, 7, 5, 2, 6, 9, 5, 11, 4]
コード例 #7
0
def test_find_max():

    tree = BinarySearchTree()
    tree.add(10)
    tree.add(5)
    tree.add(15)
    tree.add(-20)
    tree.add(1)

    assert tree.find_maximum_value() == 15
コード例 #8
0
def test_tree_one_member():
    tree = BinarySearchTree()
    tree.add('apples')
    assert tree.root.value == 'apples'
コード例 #9
0
def test_find_max_root_no_root():
  tree = BinarySearchTree()
  assert tree.find_max_val(tree.root) == None