Update test for improved engine behaviour.

The new data-flow engine means the source data is only read once.