def test_rasterize_matplotlib_basic():
im = Image.init_blank([11, 11], fill=0, n_channels=1)
im.landmarks['test'] = centre
new_im = rasterize_landmarks_2d(im, group='test', render_lines=False,
marker_style='.', marker_face_colour='r',
marker_size=2, marker_edge_width=0,
backend='matplotlib')
assert new_im.n_channels == 3
assert new_im.shape == (11, 11)
assert_allclose(new_im.pixels[:, 5, 5], [255, 0, 0])
def test_rasterize_pillow_basic():
im = Image.init_blank([11, 11], fill=0, n_channels=3)
im.landmarks['test'] = centre
new_im = rasterize_landmarks_2d(im, group='test', render_lines=False,
marker_style='s', marker_face_colour='r',
marker_size=1, marker_edge_width=0,
backend='pillow')
assert new_im.n_channels == 3
assert new_im.shape == (11, 11)
assert_allclose(new_im.pixels[0, 3:6, 3:6], 255)
def test_rasterize_pillow_basic_line():
im = Image.init_blank([11, 11], fill=0, n_channels=3)
im.landmarks['test'] = line
new_im = rasterize_landmarks_2d(im, group='test', render_lines=True,
line_width=1, line_colour='b',
marker_style='s', marker_face_colour='r',
marker_size=1, marker_edge_width=0,
backend='pillow')
assert new_im.n_channels == 3
assert_allclose(new_im.pixels[0, 1:4, 3:6], 255)
assert_allclose(new_im.pixels[0, 7:-1, 3:6], 255)
assert_allclose(new_im.pixels[2, 4:7, 4], 255)
def test_rasterize_matplotlib_basic_line():
import matplotlib
matplotlib.use('agg')
im = Image.init_blank([11, 11], fill=0, n_channels=1)
im.landmarks['test'] = line
new_im = rasterize_landmarks_2d(im, group='test', render_lines=True,
marker_style='.', marker_face_colour='r',
marker_size=2, marker_edge_width=0,
backend='matplotlib')
assert new_im.n_channels == 3
assert new_im.shape == (11, 11)
assert_allclose(new_im.pixels[0, 3, 5], [255])
assert_allclose(new_im.pixels[0, 9, 5], [255])
assert_allclose(new_im.pixels[2, 5:8, 5], [255, 255, 255])
def test_rasterize_pillow_basic():
im = Image.init_blank([11, 11], fill=0, n_channels=3)
im.landmarks['test'] = centre
new_im = rasterize_landmarks_2d(im,
group='test',
render_lines=False,
marker_style='s',
marker_face_colour='r',
marker_size=1,
marker_edge_width=0,
backend='pillow')
assert new_im.n_channels == 3
assert new_im.shape == (11, 11)
assert_allclose(new_im.pixels[0, 3:6, 3:6], 255)
def test_rasterize_matplotlib_basic():
im = Image.init_blank([11, 11], fill=0, n_channels=1)
im.landmarks['test'] = centre
new_im = rasterize_landmarks_2d(im,
group='test',
render_lines=False,
marker_style='.',
marker_face_colour='r',
marker_size=2,
marker_edge_width=0,
backend='matplotlib')
assert new_im.n_channels == 3
assert new_im.shape == (11, 11)
assert_allclose(new_im.pixels[:, 5, 5], [255, 0, 0])
def test_rasterize_pillow_basic_line():
im = Image.init_blank([11, 11], fill=0, n_channels=3)
im.landmarks['test'] = line
new_im = rasterize_landmarks_2d(im,
group='test',
render_lines=True,
line_width=1,
line_colour='b',
marker_style='s',
marker_face_colour='r',
marker_size=1,
marker_edge_width=0,
backend='pillow')
assert new_im.n_channels == 3
assert_allclose(new_im.pixels[0, 1:4, 3:6], 255)
assert_allclose(new_im.pixels[0, 7:-1, 3:6], 255)
assert_allclose(new_im.pixels[2, 4:7, 4], 255)
def test_rasterize_matplotlib_basic_line():
import matplotlib
matplotlib.use('agg')
im = Image.init_blank([11, 11], fill=0, n_channels=1)
im.landmarks['test'] = line
new_im = rasterize_landmarks_2d(im,
group='test',
render_lines=True,
marker_style='.',
marker_face_colour='r',
marker_size=2,
marker_edge_width=0,
backend='matplotlib')
assert new_im.n_channels == 3
assert new_im.shape == (11, 11)
assert_allclose(new_im.pixels[0, 3, 5], [255])
assert_allclose(new_im.pixels[0, 9, 5], [255])
assert_allclose(new_im.pixels[2, 5:8, 5], [255, 255, 255])
