def test_normalisation_stage(get_ensemble_config: dict, normalization_stage: NormalizationStage) -> None:
        """Test that model postprocessor has normalisation enabled only on tile level."""
        config = get_ensemble_config
        model = get_ensemble_model(
            config["TrainModels"]["model"],
            normalization_stage=normalization_stage,
            input_size=config["tiling"]["tile_size"],
        )

        if normalization_stage == NormalizationStage.TILE:
            assert model.post_processor.enable_normalization
        else:
            assert not model.post_processor.enable_normalization
Test that model postprocessor has normalisation enabled only on tile level.
tests/unit/pipelines/tiled_ensemble/test_helper_functions.py