def test_fill_labelled_image_different_sizes():
"""Test the numpy full labelled image building
Work with label tables, hence fixtures must be (n, n)-shaped structures
This test case corresponds to the situation where raw image width and
height are not equal.
"""
a1 = np.array([[1, 2], [3, 4]], dtype=np.int8)
a2 = np.array([[5, 6], [7, 8]], dtype=np.int8)
a3 = np.array([[9, 10], [11, 12]], dtype=np.int8)
a4 = np.array([[13, 14], [15, 16]], dtype=np.int8)
a5 = np.array([[17, 18], [19, 20]], dtype=np.int8)
a6 = np.array([[21, 22], [23, 24]], dtype=np.int8)
tiles = np.array([a1, a2, a3, a4, a5, a6])
x1, y1 = 0, 0
x2, y2 = 2, 0
x3, y3 = 0, 2
x4, y4 = 2, 2
x5, y5 = 0, 4
x6, y6 = 2, 4
coordinates = [[x1, y1], [x2, y2], [x3, y3], [x4, y4], [x5, y5], [x6, y6]]
e1 = np.array(
[
[1, 2, 5, 6],
[3, 4, 7, 8],
[9, 10, 13, 14],
[11, 12, 15, 16],
[17, 18, 21, 22],
[19, 20, 23, 24],
],
dtype=np.int8,
)
e2 = postprocess.fill_labelled_image(tiles, coordinates, 2, 4, 6)
assert np.all(e1 == e2)
def test_fill_labelled_image():
"""Test the numpy full labelled image building
Work with pixel tables, hence fixtures must be (n, n, 3)-shaped structures
"""
a1 = np.array([[1, 2], [3, 4]], dtype=np.int8)
a2 = np.array([[5, 6], [7, 8]], dtype=np.int8)
a3 = np.array([[9, 10], [11, 12]], dtype=np.int8)
a4 = np.array([[13, 14], [15, 16]], dtype=np.int8)
tiles = np.array([a1, a2, a3, a4])
x1, y1 = 0, 0
x2, y2 = 2, 0
x3, y3 = 0, 2
x4, y4 = 2, 2
coordinates = [[x1, y1], [x2, y2], [x3, y3], [x4, y4]]
e1 = np.array(
[[1, 2, 5, 6], [3, 4, 7, 8], [9, 10, 13, 14], [11, 12, 15, 16]],
dtype=np.int8)
e2 = postprocess.fill_labelled_image(tiles, coordinates, 2, 4)
assert np.all(e1 == e2)