Updated tests for complete PITO example