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) == []