Design and analysis of a wake model for spatially heterogeneous flow