Update tests for source and sink simulations.