def test_default_size(self):
sl = square_letterboxed(self.img)
assert sl.shape[0] == SQUARE_LETTERBOXED_DEFAULT_SIZE
assert sl.shape[1] == SQUARE_LETTERBOXED_DEFAULT_SIZE
def test_fill_value_27(self):
sl = square_letterboxed(self.img, size=5, fill_value=27)
assert np.allclose(sl, SL_27)
def test_size(self):
sl = square_letterboxed(self.img, size=20)
assert sl.shape == (20, 20)
def test_fill_value_mean(self):
sl = square_letterboxed(self.img, size=5, fill_value='mean')
assert np.allclose(sl, SL_MEAN)