def test_iou():
bb_a = BBox(0, 2, 0, 2)
bb_b = BBox(3, 5, 0, 1)
bb_c = BBox(0, 1, 3, 4)
assert iou(bb_a, bb_b) == 0
assert iou(bb_a, bb_c) == 0
assert iou(bb_b, bb_c) == 0
assert iou(bb_a, bb_a) == 1
bb_d = BBox(1, 3, 1, 3)
assert iou(bb_a, bb_d) == get_intersection_area(bb_a, bb_d) / get_union_area(
bb_a, bb_d
)
def test_get_intersection_area():
bb_a = BBox(0, 2, 0, 2)
bb_b = BBox(3, 5, 0, 1)
bb_c = BBox(0, 1, 3, 4)
assert get_intersection_area(bb_a, bb_b) == 0
assert get_intersection_area(bb_a, bb_c) == 0
assert get_intersection_area(bb_b, bb_c) == 0
bb_d = BBox(1, 3, 1, 3)
assert get_intersection_area(bb_a, bb_d) == 1
assert get_intersection_area(bb_d, bb_a) == 1
assert get_intersection_area(bb_a, bb_a) == 4
bb_e = BBox(0, 5, 0, 5)
assert get_intersection_area(bb_e, bb_a) == 4
assert get_intersection_area(bb_e, bb_b) == 2
def test_get_union_area():
bb_a = BBox(0, 2, 0, 2)
bb_b = BBox(3, 5, 0, 1)
bb_c = BBox(0, 1, 3, 4)
assert get_union_area(bb_a, bb_b) == bb_a.area + bb_b.area
assert get_union_area(bb_a, bb_c) == bb_a.area + bb_c.area
assert get_union_area(bb_b, bb_c) == bb_b.area + bb_c.area
assert get_union_area(bb_a, bb_a) == bb_a.area
bb_d = BBox(1, 3, 1, 3)
assert get_union_area(bb_a, bb_d) == bb_a.area + bb_d.area - get_intersection_area(
bb_a, bb_d
)
bb_e = BBox(0, 5, 0, 5)
assert get_union_area(bb_e, bb_b) == bb_e.area