def test_load_result(self): anxcor = Anxcor() anxcor.set_window_length(120.0) times = anxcor.get_starttimes(starttime_stamp, endtime_stamp, 0.5) bank = WavebankWrapper(source_dir) anxcor.add_dataset(bank, 'nodals') original_result = anxcor.process(times) anxcor.save_result(original_result, target_dir) final_result = anxcor.load_result(target_dir) assert len(final_result.to_array().squeeze().data.shape) == 5
def test_save_result(self): anxcor = Anxcor() anxcor.set_window_length(120.0) times = anxcor.get_starttimes(starttime_stamp, endtime_stamp, 0.5) bank = WavebankWrapper(source_dir) anxcor.add_dataset(bank, 'nodals') result = anxcor.process(times) anxcor.save_result(result, target_dir) num_nc = _how_many_fmt(target_dir, format='.nc') assert num_nc == 1