def test_ensemble_datamodule_img_size(get_ensemble_config: dict, get_tiler: EnsembleTiler) -> None:
        """Test that datamodule is created and has correct collate function."""
        config = get_ensemble_config
        tiler = get_tiler
        datamodule = get_ensemble_datamodule(config, config["tiling"]["image_size"], tiler, (0, 0))

        data = datamodule.test_data[0]

        # check that transforms were correctly set to dataset
        assert list(data.image.shape[1:]) == config["tiling"]["image_size"]
Test that datamodule is created and has correct collate function.
tests/unit/pipelines/tiled_ensemble/test_helper_functions.py