TST: added the tests to save_output
TST: modified the test for output_save