Change the default colormap for exponentiated objects to match that for un-exponentiated objects, and also update test_fancy_ptycho to show all the plots (not just plot level 2)

This commit is contained in:
allevitan committed 2026-03-24 16:55:56 +01:00
1 parent ad2b09f2f0
commit 603486e824
2 files changed
+7 -3

No files matched your search

+5 -2
View File
@@ -1004,7 +1004,8 @@ class FancyPtycho(CDIModel):
basis=self.obj_basis,
additional_axis_labels=['Mode #',],
units=self.units,
cmap='cividis'),
cmap='cividis',
),
'condition': lambda self: self.exponentiate_obj,
},
{
@@ -1015,7 +1016,9 @@ class FancyPtycho(CDIModel):
fig=fig,
basis=self.obj_basis,
additional_axis_labels=['Mode #',],
units=self.units),
units=self.units,
cmap='viridis_r',
),
'condition': lambda self: self.exponentiate_obj,
},
{
+2 -1
View File
@@ -69,7 +69,8 @@ def test_lab_ptycho(lab_ptycho_cxi, reconstruction_device, show_plot):
units='mm',
obj_view_crop=-50,
use_qe_mask=True, # test this in the case where no qe mask is defined
panel_plot_mode=True, # test with panel plot mode
panel_plot_mode=True, # test with panel plot mode,
plot_level=4, # test with all plots
)
print('Running reconstruction on provided reconstruction_device,',