def test_three_horizontal(self):
feature = HaarLikeFeature(FeatureType.THREE_HORIZONTAL, (0, 0), 24, 24, 100000, 1)
left_area = ii.sum_region(self.int_img, (0, 0), (8, 24))
middle_area = ii.sum_region(self.int_img, (8, 0), (16, 24))
right_area = ii.sum_region(self.int_img, (16, 0), (24, 24))
expected = 1 if feature.threshold * feature.polarity > left_area - middle_area + right_area else 0
assert feature.get_vote(self.int_img) == expected
def test_three_vertical(self):
feature = HaarLikeFeature(FeatureType.THREE_VERTICAL, (0, 0), 24, 24, 100000, 1)
left_area = ii.sum_region(self.int_img, (0, 0), (24, 8))
middle_area = ii.sum_region(self.int_img, (0, 8), (24, 16))
right_area = ii.sum_region(self.int_img, (0, 16), (24, 24))
expected = 1 if feature.threshold * feature.polarity > left_area - middle_area + right_area else 0
assert feature.get_vote(self.int_img) == expected
def test_four(self):
feature = HaarLikeFeature(FeatureType.THREE_HORIZONTAL, (0, 0), 24, 24, 100000, 1)
top_left_area = ii.sum_region(self.int_img, (0, 0), (12, 12))
top_right_area = ii.sum_region(self.int_img, (12, 0), (24, 12))
bottom_left_area = ii.sum_region(self.int_img, (0, 12), (12, 24))
bottom_right_area = ii.sum_region(self.int_img, (12, 12), (24, 24))
expected = 1 if feature.threshold * feature.polarity > top_left_area - top_right_area - bottom_left_area + bottom_right_area else 0
assert feature.get_vote(self.int_img) == expected
def test_two_vertical_fail(self):
feature = HaarLikeFeature(FeatureType.TWO_VERTICAL, (0, 0), 24, 24, 100000, 1)
left_area = ii.sum_region(self.int_img, (0, 0), (24, 12))
right_area = ii.sum_region(self.int_img, (0, 12), (24, 24))
expected = 1 if feature.threshold * -1 > left_area - right_area else 0
assert feature.get_vote(self.int_img) != expected