def test_downsample_brain_rescale():
downsampler = pixella.Downsampler('brain.png')
from PIL import Image
img = downsampler.get_new_image(rescale=True, sample_rate=10)
result = downsampler.downsample(img, sample_rate=10, rescale=True)
comparison = Image.open('rescaled_brain.png')
assert list(result.getdata()) == list(comparison.getdata())
def test_get_pixels_as_rows():
file = 'wbrgb.png'
downsampler = pixella.Downsampler(file)
from PIL import Image
img = Image.open(file)
result = downsampler.get_pixels_as_rows(img)
desired_values = [[(255, 255, 255), (0, 0, 0), (255, 0, 0), (0, 255, 0),
(0, 0, 255)],
[(0, 0, 255), (0, 0, 255), (0, 0, 255), (0, 0, 255),
(0, 0, 255)]]
assert result == desired_values
def test_get_new_image_rescale():
downsampler = pixella.Downsampler('100x100.png')
result = downsampler.get_new_image(rescale=True, sample_rate=20)
w, h = result.size
assert w == 6
assert h == 6
def test_get_new_image_no_rescale():
downsampler = pixella.Downsampler('100x100.png')
result = downsampler.get_new_image()
w, h = result.size
assert w == 100
assert h == 100