Esempio n. 1
0
 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
Esempio n. 2
0
 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
Esempio n. 3
0
 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
Esempio n. 4
0
 def test_init(self):
     tree = KDTree(1)
     assert tree.root is None
     assert tree.size == 0
     assert tree.is_empty() is True