Skip to content

Commit 28c380d

Browse files
committed
More test cases
1 parent 291d181 commit 28c380d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/test_prototype_transforms_functional.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,7 @@ def pad_segmentation_mask():
382382

383383
@register_kernel_info_from_sample_inputs_fn
384384
def center_crop_bounding_box():
385-
for bounding_box, output_size in itertools.product(make_bounding_boxes(), [(24, 12), (16, 18)]):
385+
for bounding_box, output_size in itertools.product(make_bounding_boxes(), [(24, 12), [16, 18], [46, 48], [12]]):
386386
yield SampleInput(
387387
bounding_box, format=bounding_box.format, output_size=output_size, image_size=bounding_box.image_size
388388
)
@@ -1098,7 +1098,7 @@ def parse_padding():
10981098
@pytest.mark.parametrize("device", cpu_and_gpu())
10991099
@pytest.mark.parametrize(
11001100
"output_size",
1101-
[(18, 18), [18, 15], (16, 19), [12]],
1101+
[(18, 18), [18, 15], (16, 19), [12], [46, 48]],
11021102
)
11031103
def test_correctness_center_crop_bounding_box(device, output_size):
11041104
def _compute_expected_bbox(bbox, output_size_):

0 commit comments

Comments
 (0)