Measure production from individual stages in multi-stage wells