Add tests for progress callback