Spaces:
Runtime error
Runtime error
FrancisGOS
commited on
Commit
•
4ad0e47
1
Parent(s):
a485c33
<Fix>: fix test
Browse files
app/custom_mmcv/test_custom_mmcv.py
CHANGED
@@ -30,22 +30,22 @@ class TestCustomMMCV():
|
|
30 |
image = mmcv.imread('demo.jpg')
|
31 |
bboxes = np.ones((1,5))
|
32 |
labels = np.zeros(1, dtype=np.int32)
|
33 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red", text_color='red'
|
34 |
assert (result[1,1,:] == (0,0,255)).all()
|
35 |
with pytest.raises(AssertionError):
|
36 |
bboxes = np.ones((1,3))
|
37 |
labels = np.zeros(1, dtype=np.int32)
|
38 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red"
|
39 |
with pytest.raises(AssertionError):
|
40 |
bboxes = np.ones((1,7))
|
41 |
labels = np.zeros(1, dtype=np.int32)
|
42 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red"
|
43 |
with pytest.raises(AssertionError):
|
44 |
bboxes = np.ones((1,5))
|
45 |
labels = np.zeros(4, dtype=np.int32)
|
46 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red"
|
47 |
with pytest.raises(AssertionError):
|
48 |
bboxes = np.ones((2,5))
|
49 |
labels = np.zeros(1, dtype=np.int32)
|
50 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red"
|
51 |
|
|
|
30 |
image = mmcv.imread('demo.jpg')
|
31 |
bboxes = np.ones((1,5))
|
32 |
labels = np.zeros(1, dtype=np.int32)
|
33 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red", text_color='red')
|
34 |
assert (result[1,1,:] == (0,0,255)).all()
|
35 |
with pytest.raises(AssertionError):
|
36 |
bboxes = np.ones((1,3))
|
37 |
labels = np.zeros(1, dtype=np.int32)
|
38 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red")
|
39 |
with pytest.raises(AssertionError):
|
40 |
bboxes = np.ones((1,7))
|
41 |
labels = np.zeros(1, dtype=np.int32)
|
42 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red")
|
43 |
with pytest.raises(AssertionError):
|
44 |
bboxes = np.ones((1,5))
|
45 |
labels = np.zeros(4, dtype=np.int32)
|
46 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red")
|
47 |
with pytest.raises(AssertionError):
|
48 |
bboxes = np.ones((2,5))
|
49 |
labels = np.zeros(1, dtype=np.int32)
|
50 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red")
|
51 |
|