def test_check1(): """Test check without arguments""" bounding_boxes, confidences, labels, _ = setup_test_case() bb = BoundingBoxArray(bounding_boxes=bounding_boxes, confidences=confidences, labels=labels) bb.check()
def smoke_test_setup(): bounding_boxes, confidences, labels, _ = setup_test_case() bb = BoundingBoxArray(bounding_boxes=bounding_boxes, confidences=confidences, labels=labels) bb.check() metric = DefaultIntersectionOverTheUnion(threshold=0.35, direction="lt") selector = RandomSelector() return bb, metric, selector
def test_check2(): """Test check with arguments""" bounding_boxes, confidences, labels, _ = setup_test_case() bounding_box, confidence, label = one_additional() bb = BoundingBoxArray(bounding_boxes=bounding_boxes, confidences=confidences, labels=labels) bb.check(bounding_box=bounding_box, confidence=confidence, label=label)
def test_check7(): """Test check without arguments - fail bb range""" bounding_boxes, confidences, labels, _ = setup_test_case() bounding_boxes[0, 0, 0] = 1.01 bb = BoundingBoxArray(bounding_boxes=bounding_boxes, confidences=confidences, labels=labels) with pytest.raises(ValueError): bb.check()
def test_check5(): """Test check without arguments - warn""" bounding_boxes, confidences, labels, _ = setup_test_case() labels = labels.astype(np.float64) bb = BoundingBoxArray(bounding_boxes=bounding_boxes, confidences=confidences, labels=labels) with pytest.warns(SyntaxWarning): bb.check()
def test_check6(): """Test check with arguments - warn""" bounding_boxes, confidences, labels, _ = setup_test_case() bounding_box, confidence, label = one_additional() bounding_box = bounding_box.astype(np.int64) bb = BoundingBoxArray(bounding_boxes=bounding_boxes, confidences=confidences, labels=labels) with pytest.warns(SyntaxWarning): bb.check(bounding_box=bounding_box, confidence=confidence, label=label)
def test_check9(): """Test check without arguments - fail bb shape len""" bounding_boxes, confidences, labels, _ = setup_test_case() bounding_boxes = np.array( ( ((0.04, 0.19, 0.12), (0.14, 0.29, 0.81)), ((0.11, 0.15, 0.02), (0.21, 0.25, 0.14)), ), dtype=np.float64, ) bb = BoundingBoxArray(bounding_boxes=bounding_boxes, confidences=confidences, labels=labels) with pytest.raises(ValueError): bb.check()