def test_uni_plot_ax_bis(self):
# two plots same figures
aplot_1 = APlot()
aplot_1.uni_plot(0, self.xx, self.yy + 5)
aplot_1.uni_plot_ax_bis(0, self.xx, np.exp(self.xx))
self.image_name = "image_plot_bis"
self.check_plot()
def test_show_legend_for_two_axis(self):
yy = np.abs(np.cos(self.xx)) + 1
# second, try the same as before where the second axis is also setting another characteristic.
aplot = APlot()
dict_plot1 = {'title': 'my title1'}
dict_plot2 = {'title': 'my title2'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot1)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot2)
aplot.show_legend()
self.image_name = "image_legend_two_axis"
self.check_plot()
def test_set_dict_ax_and_bis_each_parameter_only_first_two_axis_same_graph(
self):
# first, trying every simple possibility for dict_ax
yy = np.abs(np.cos(self.xx)) + 1
with self.subTest('title'):
aplot = APlot()
dict_plot = {'title': 'my title'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_title"
self.check_plot()
with self.subTest('xlabel'):
aplot = APlot()
dict_plot = {'xlabel': 'my x label'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_xlabel"
self.check_plot()
with self.subTest('ylabel'):
aplot = APlot()
dict_plot = {'ylabel': 'my y label'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_ylabel"
self.check_plot()
with self.subTest('xscale'):
aplot = APlot()
dict_plot = {'xscale': 'log'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_xscale_only_principal_axis"
self.check_plot()
with self.subTest('yscale left'):
aplot = APlot()
dict_plot = {'yscale': 'log'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_yscale"
self.check_plot()
with self.subTest('xint'):
aplot = APlot()
dict_plot = {'xint': True}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 1 / 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_xint"
self.check_plot()
with self.subTest('xint_yint'):
aplot = APlot()
dict_plot = {'xint': True, 'yint': True}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 1 / 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_xyint"
self.check_plot()
for i in range(11):
with self.subTest('parameter', i=i):
aplot = APlot()
dict_plot = {
'parameters': ['A', 3, 5] * i,
'name_parameters': ['A', '$\sigma$', '$\\rho$'] * i
}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
self.image_name = f"image_set_dict_ax_parameter_{i}"
self.check_plot()
# with self.subTest('parameters not good length'):
# aplot = APlot()
#
# dict_plot = {
# 'parameters': ['A', 3, 5, 10, 42], 'name_parameters': ['A']}
#
# aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
# aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
with self.subTest('xlim'):
aplot = APlot()
dict_plot = {'xlim': [0, 0.5]}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_xlim"
self.check_plot()
with self.subTest('ylim'):
aplot = APlot()
dict_plot = {'ylim': [1.2, 2.4]}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1)
self.image_name = "image_set_dict_ax_ylim"
self.check_plot()
def test_set_dict_ax_and_bis_each_parameter_both_two_axis_same_graph(self):
yy = np.abs(np.cos(self.xx)) + 1
# second, try the same as before where the second axis is also setting another characteristic.
with self.subTest('two different titles'):
aplot = APlot()
dict_plot1 = {'title': 'my title1'}
dict_plot2 = {'title': 'my title2'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot1)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot2)
self.image_name = "image_set_dict_ax_title_2"
self.check_plot()
with self.subTest('two xlabels'):
aplot = APlot()
dict_plot1 = {'xlabel': 'my x label1'}
dict_plot2 = {'xlabel': 'my x label2'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot1)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot2)
self.image_name = "image_set_dict_ax_xlabel_2"
self.check_plot()
with self.subTest('two ylabels'):
aplot = APlot()
dict_plot1 = {'ylabel': 'my y label1'}
dict_plot2 = {'ylabel': 'my y label2'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot1)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot2)
self.image_name = "image_set_dict_ax_ylabel_2"
self.check_plot()
with self.subTest('xscale same both axis'):
aplot = APlot()
dict_plot = {'xscale': 'log'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot)
self.image_name = "image_set_dict_ax_xscale_both_same_scale"
self.check_plot()
with self.subTest('yscale same both axis'):
aplot = APlot()
dict_plot = {'yscale': 'log'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot)
self.image_name = "image_set_dict_ax_yscale_2_same"
self.check_plot()
with self.subTest('xscale different'):
aplot = APlot()
dict_plot2 = {'xscale': 'log'}
dict_plot = {'xscale': 'linear'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot2)
self.image_name = "image_set_dict_ax_xscale_2_different_scales"
self.check_plot()
with self.subTest('yscale different'):
aplot = APlot()
dict_plot = {'yscale': 'log'}
dict_plot2 = {'yscale': 'linear'}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot2)
self.image_name = "image_set_dict_ax_yscale_2_different"
self.check_plot()
# with self.subTest('xint same'):
# aplot = APlot()
#
# dict_plot = {'xint': True}
#
# aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
# aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1, dict_ax=dict_plot)
#
# plt.savefig("image_reference_test_plot/test_image_set_dict_ax_xint_2_same.png")
#
# with self.subTest('xint_yint same'):
# aplot = APlot()
#
# dict_plot = {'xint': True, 'yint': True}
#
# aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
# aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1, dict_ax=dict_plot)
#
# plt.savefig("image_reference_test_plot/test_image_set_dict_ax_xyint_same_2.png")
# with self.subTest('xint different'):
# aplot = APlot()
#
# dict_plot = {'xint': True}
# dict_plot2 = {'xint': False}
#
# aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
# aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1, dict_ax=dict_plot2)
#
# plt.savefig("image_reference_test_plot/test_image_set_dict_ax_xint_2_different_2.png")
#
# with self.subTest('xint_yint different'):
# aplot = APlot()
#
# dict_plot = {'xint': True, 'yint': True}
# dict_plot2 = {'xint': True, 'yint': False}
#
# aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot)
# aplot.uni_plot_ax_bis(0, self.xx, 3 * np.sin(self.xx) + 1, dict_ax=dict_plot2)
#
# plt.savefig("image_reference_test_plot/test_image_set_dict_ax_xyint_different_2.png")
with self.subTest('xlim'):
aplot = APlot()
dict_plot1 = {'xlim': [0, 0.5]}
dict_plot2 = {'xlim': [0, 1.5]}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot1)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot2)
self.image_name = "image_set_dict_ax_xlim_2"
self.check_plot()
with self.subTest('ylim'):
aplot = APlot()
dict_plot1 = {'ylim': [1, 5]}
dict_plot2 = {'ylim': [1, 15]}
aplot.uni_plot(0, self.xx, yy, dict_ax=dict_plot1)
aplot.uni_plot_ax_bis(0,
self.xx,
3 * np.sin(self.xx) + 1,
dict_ax=dict_plot2)
self.image_name = "image_set_dict_ax_ylim_2"
self.check_plot()
