def test_get_index_median(self): """Test if the median returned is correct""" # ODD CASE self.assertEqual( KdTree._get_index_median(self.kd_tree_odd, self.points_odd), 3) # EVEN CASE self.assertEqual( KdTree._get_index_median(self.kd_tree_even, self.points_even), 2)