def test_ensemble_model(get_ensemble_config: dict) -> None:
        """Test that model is successfully created with correct input shape."""
        config = get_ensemble_config
        model = get_ensemble_model(
            config["TrainModels"]["model"],
            normalization_stage=config["normalization_stage"],
            input_size=config["tiling"]["tile_size"],
        )
        if hasattr(model.model, "input_size"):
            # if model has fixed input_size, it should be set to tile_size
            assert model.model.input_size == tuple(config["tiling"]["tile_size"])
Test that model is successfully created with correct input shape.
tests/unit/pipelines/tiled_ensemble/test_helper_functions.py