def plot_history(): directory = get_directory() lateral, axial = collect_limits(directory) print(lateral, axial) subs = get_subdirectories(directory) frame_psf = pd.DataFrame() frame_pow = pd.DataFrame() for folder in subs: data_psf = collect_data_psf(os.path.join(directory, folder)) data_pow = collect_data_pow(os.path.join(directory, folder)) if isinstance(data_pow, pd.DataFrame): frame_pow = append_data_pow(data_pow, frame_pow, folder) if isinstance(data_psf, pd.DataFrame): frame_psf = append_data_psf(data_psf, frame_psf, folder) plot_psf(frame_psf, directory, lateral, axial) plot_pow(frame_pow, directory) return
def test_empty_folder_pow(): folder = [] assert collect_data_psf(folder) == []
def test_return_type_pow(): folder = './test_data' assert isinstance(collect_data_psf(folder), pd.DataFrame)
def test_return_size_psf(): folder = './test_data' print(collect_data_psf(folder).shape) assert collect_data_psf(folder).shape == (9, 5)
def test_no_file_psf(): folder = '.' assert collect_data_psf(folder) == []