def test_tile_mosaic_position(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) tile_id = 0 np.testing.assert_equal([0], tiler.get_tile_mosaic_position(tile_id)) np.testing.assert_equal([0], tiler.get_tile_mosaic_position( tile_id, with_channel_dim=True)) np.testing.assert_equal([0], tiler2.get_tile_mosaic_position(tile_id)) np.testing.assert_equal([0, 0], tiler2.get_tile_mosaic_position( tile_id, with_channel_dim=True)) tile_id = len(tiler) - 1 np.testing.assert_equal([9], tiler.get_tile_mosaic_position(tile_id)) np.testing.assert_equal([9], tiler.get_tile_mosaic_position( tile_id, with_channel_dim=True)) np.testing.assert_equal([9], tiler2.get_tile_mosaic_position(tile_id)) np.testing.assert_equal([0, 9], tiler2.get_tile_mosaic_position( tile_id, with_channel_dim=True)) with self.assertRaises(IndexError): tiler.get_tile_mosaic_position(-1) with self.assertRaises(IndexError): tiler.get_tile_mosaic_position(len(tiler))