def test_resize_keep_aspect_ratio_horizontally_downsample(): img = np.zeros((4, 8)) color = 250 result_img = np.zeros((4, 4)) result_img[[0, -1]] = color img2 = resize_if_needed(img, (4, 4), True, color) assert np.array_equal(result_img, img2)
def test_resize_keep_aspect_ratio_vertically_upsample(): img = np.zeros((8, 4)) color = 250 result_img = np.zeros((10, 4)) result_img[[0, -1]] = color img2 = resize_if_needed(img, (4, 10), True, color) assert np.array_equal(result_img, img2)
def test_resize_keep_aspect_ratio_mixed_downsample(): img = np.zeros((4, 8)) color = 250 result_img = np.zeros((8, 4)) result_img[[0, 1, 2, -3, -2, -1]] = color img2 = resize_if_needed(img, (4, 8), True, color) assert np.array_equal(result_img, img2)
def test_resize_keep_last_dim(): img = np.zeros((8, 4)) img = img.reshape((8, 4, 1)) img2 = resize_if_needed(img, (8, 4)) assert img2.shape == (4, 8, 1)
def test_resize_handle_dimensions_correctly(): img = np.zeros((8, 4)) img2 = resize_if_needed(img, (8, 4)) assert img2.shape == (4, 8)
def test_resize_avoid_useless_resize(): img = np.zeros((4, 4)) img2 = resize_if_needed(img, img.shape) assert img is img2