Refactor and improve framework for testing model update output