TST: modified the tests for save_output