def test_collate_tile_shape(get_ensemble_config: dict, get_datamodule: AnomalibDataModule) -> None:
        """Test that collate function successfully tiles the image."""
        config = get_ensemble_config
        # datamodule with tile collater
        datamodule = get_datamodule

        tile_w, tile_h = config["tiling"]["tile_size"]

        batch = next(iter(datamodule.train_dataloader()))
        assert batch.image.shape[1:] == (3, tile_w, tile_h)
        assert batch.gt_mask.shape[1:] == (tile_w, tile_h)
Test that collate function successfully tiles the image.
tests/unit/pipelines/tiled_ensemble/test_tiler.py