def test_mosaic_shape(self):
tile_size = 10
tiler = Tiler(data_shape=self.data.shape, tile_shape=(tile_size, ))
tiler2 = Tiler(data_shape=(3, ) + self.data.shape,
tile_shape=(
3,
tile_size,
),
channel_dimension=0)
tiler3 = Tiler(data_shape=(9, ) + self.data.shape,
tile_shape=(
3,
tile_size,
),
channel_dimension=0)
tiler4 = Tiler(data_shape=(9, ) + self.data.shape,
tile_shape=(
3,
tile_size,
))
np.testing.assert_equal([10], tiler.get_mosaic_shape())
np.testing.assert_equal([10],
tiler.get_mosaic_shape(with_channel_dim=True))
np.testing.assert_equal([10], tiler2.get_mosaic_shape())
np.testing.assert_equal([1, 10],
tiler2.get_mosaic_shape(with_channel_dim=True))
np.testing.assert_equal([10], tiler3.get_mosaic_shape())
np.testing.assert_equal([1, 10],
tiler3.get_mosaic_shape(with_channel_dim=True))
np.testing.assert_equal([3, 10], tiler4.get_mosaic_shape())
np.testing.assert_equal([3, 10],
tiler4.get_mosaic_shape(with_channel_dim=True))
