def test_plot_chain(self):
X2_list = [1, 1, 2]
pos_list = [[1, 0], [2, 0], [3, 0]]
vel_list = [[-1, 0], [0, 0], [1, 0]]
param_list = ['test1', 'test2']
chain = X2_list, pos_list, vel_list, None
output_plots.plot_chain(chain=chain, param_list=param_list)
plt.close()
arrow_size=0.02,
cmap_string="gist_heat")
lensPlotG = LensModelPlot(kwargs_dataG,
kwargs_psfG,
kwargs_numerics,
kwargs_modelG,
lens_result,
source_result,
lens_light_resultG,
ps_result,
arrow_size=0.02,
cmap_string="gist_heat")
for i in range(len(chain_list)):
if len(param_list[i]) > 0:
f, axes = out_plot.plot_chain(chain_list[i], param_list[i])
f, axes = plt.subplots(1, 3, figsize=(16, 8), sharex=False, sharey=False)
plt.title('I band')
lensPlotI.data_plot(ax=axes[0])
lensPlotI.model_plot(ax=axes[1])
lensPlotI.normalized_residual_plot(ax=axes[2], v_min=-6, v_max=6)
f.tight_layout()
#f.subplots_adjust(left=None, bottom=None, right=None, top=None, wspace=0., hspace=0.05)
#plt.show()
plt.savefig('E:\\twosersicresults\\' + name[0:8] + 'Ibandplot2.png')
f, axes = plt.subplots(1, 3, figsize=(16, 8), sharex=False, sharey=False)
plt.title('R band')
lensPlotR.data_plot(ax=axes[0])