def test_init_with_list_dim_3(self): tree = KDTree(3, [(1, 2, 3), (0, 1, 4), (2, 4, 3)]) assert tree.root.data == (1, 2, 3) assert tree.root.left.data == (0, 1, 4) assert tree.root.right.data == (2, 4, 3) assert tree.size == 3 assert tree.is_empty() is False
def test_init_with_list_dim_2(self): tree = KDTree(2, [(1, 1), (3, 3), (2, 2)]) assert tree.root.data == (1, 1) assert tree.root.left == None assert tree.root.right.data == (3, 3) assert tree.root.right.left.data == (2, 2) assert tree.size == 3 assert tree.is_empty() is False
def test_init_with_larger_list_dim_5(self): tree = KDTree(5, [(1, 2, 3, 4, 5), (0, 1, 4, 1, 2), (2, 4, 3, 6, 7), (9, 8, 10, 7, 3), (-1, 0, 0, 14, 15)]) assert tree.root.data == (1, 2, 3, 4, 5) assert tree.root.left.data == (0, 1, 4, 1, 2) assert tree.root.right.data == (2, 4, 3, 6, 7) assert tree.root.right.right.data == (9, 8, 10, 7, 3) assert tree.root.left.left.data == (-1, 0, 0, 14, 15) assert tree.size == 5 assert tree.is_empty() is False
def test_init(self): tree = KDTree(1) assert tree.root is None assert tree.size == 0 assert tree.is_empty() is True