def test_3d_three_channel_input():
shape = rand_shape_3d()
shape = (3,) + shape[1:]
data = rand_ndarray(shape, 'default')
grid = make_image_grid(data)
assert grid.ndim == 3
assert grid.shape[0] == 3
assert same(grid.asnumpy(), data.asnumpy())
def test_3d_single_channel_input():
shape = rand_shape_3d(dim0=1)
data = rand_ndarray(shape, 'default')
assert data.shape[0] == 1 # single channel
grid = make_image_grid(data)
assert grid.ndim == 3
assert grid.shape[0] == 3
assert same(grid[0].asnumpy(), grid[1].asnumpy())
assert same(grid[0].asnumpy(), grid[2].asnumpy())
assert same(grid[0:1].asnumpy(), data.asnumpy())
def test_2d_input():
shape = rand_shape_2d()
data = rand_ndarray(shape, 'default')
grid = make_image_grid(data)
assert grid.ndim == 3
assert grid.shape[0] == 3
assert grid.shape[1:] == data.shape
assert same(grid[0].asnumpy(), grid[1].asnumpy())
assert same(grid[0].asnumpy(), grid[2].asnumpy())
assert same(grid[0].asnumpy(), data.asnumpy())
def test_4d_multiple_batch_input():
shape_list = list(rand_shape_nd(4))
shape_list[0] = 10
num_channels = [1, 3]
for c in num_channels:
shape_list[1] = c
shape = tuple(shape_list)
data = rand_ndarray(shape, 'default')
grid = make_image_grid(data)
assert grid.ndim == 3
assert grid.shape[0] == 3
def test_4d_single_batch_single_channel_input():
shape = list(rand_shape_nd(4))
shape[0] = 1
shape[1] = 1
shape = tuple(shape)
data = rand_ndarray(shape, 'default')
grid = make_image_grid(data)
assert grid.ndim == 3
assert grid.shape[0] == 3
assert same(grid[0].asnumpy(), grid[1].asnumpy())
assert same(grid[0].asnumpy(), grid[2].asnumpy())
assert same(grid[0].reshape(data.shape).asnumpy(), data.asnumpy())
