def test__determine_limits():
o = Pyposmat2DDensityPlot(config=config_fn, data=data_fn)
for qoi_name in o.configuration.qoi_names:
lim_min, lim_max = o.determine_limits(qoi_name)
assert isinstance(lim_min, float)
assert isinstance(lim_max, float)
assert lim_min < lim_max
def test____init____all_args_as_str():
o = Pyposmat2DDensityPlot(config=config_fn, data=data_fn)
assert isinstance(o, Pyposmat2DDensityPlot)
assert isinstance(o.configuration, PyposmatConfigurationFile)
assert isinstance(o.data, PyposmatDataFile)
assert o.fig is None
assert o.ax is None
def test____init____data_as_str():
o = Pyposmat2DDensityPlot(data=data_fn)
assert isinstance(o, Pyposmat2DDensityPlot)
assert o.configuration is None
assert isinstance(o.data, PyposmatDataFile)
assert o.fig is None
assert o.ax is None
def test____init____config_as_obj():
o = Pyposmat2DDensityPlot(config=PyposmatConfigurationFile())
assert isinstance(o, Pyposmat2DDensityPlot)
assert isinstance(o.configuration, PyposmatConfigurationFile)
assert o.data is None
assert o.fig is None
assert o.ax is None
def test____init____no_args():
o = Pyposmat2DDensityPlot()
assert isinstance(o, Pyposmat2DDensityPlot)
assert o.configuration is None
assert o.data is None
assert o.fig is None
assert o.ax is None
def test__determine_y_limits():
o = Pyposmat2DDensityPlot(config=config_fn, data=data_fn)
o.y_name = o.configuration.qoi_names[1]
o.determine_y_limits()
assert isinstance(o.y_limits[0], float)
assert isinstance(o.y_limits[1], float)
from pypospack.pyposmat.data import PyposmatConfigurationFile
from pypospack.pyposmat.data import PyposmatDataFile
from pypospack.pyposmat.visualization import Pyposmat2DDensityPlot
data_directory = os.path.join(pypospack.utils.get_pypospack_root_directory(),
'data', 'Si__sw__data',
'pareto_optimization_unconstrained')
plot_directory = 'qoi_2d_density_plots'
if os.path.isdir(plot_directory):
shutil.rmtree(plot_directory)
os.mkdir(plot_directory)
o_config = PyposmatConfigurationFile()
o_config.read(filename=os.path.join(data_directory, 'pyposmat.config.in'))
max_iteration = o_config.n_iterations
o_data = PyposmatDataFile()
o_data.read(filename=os.path.join(data_directory, 'pyposmat.kde.{}.out'.format(
max_iteration)))
for i, name_1 in enumerate(o_config.qoi_names):
for j, name_2 in enumerate(o_config.qoi_names):
if i < j:
plot_fn = '{}__{}'.format(name_1, name_2).replace('.', '_')
plot_fn += ".png"
plot_fn = os.path.join(plot_directory, plot_fn)
o = Pyposmat2DDensityPlot(config=o_config, data=o_data)
o.plot(x_name=name_1, y_name=name_2)
o.fig.savefig(plot_fn, dpi=1300)
def dev__determine_limits():
o = Pyposmat2DDensityPlot(config=config_fn, data=data_fn)
o.create_subplots()
for qoi_name in o.configuration.qoi_names:
print(qoi_name)
print(o.determine_limits(qoi_name))
def test__create_subplots__no_args():
o = Pyposmat2DDensityPlot(config=config_fn, data=data_fn)
o.create_subplots()
assert isinstance(o.fig, plt.Figure)
assert isinstance(o.ax, plt.Axes)
def test__plot():
o = Pyposmat2DDensityPlot(config=config_fn, data=data_fn)
x_name = o.configuration.qoi_names[0]
y_name = o.configuration.qoi_names[1]
o.plot(x_name=x_name, y_name=y_name)
plt.show()
def dev__determine_y_limits():
o = Pyposmat2DDensityPlot(config=config_fn, data=data_fn)
o.y_name = o.configuration.qoi_names[1]
o.determine_y_limits()
print(o.y_limits)