bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 11:41:49 +02:00
Code Activity

test: test_extreme.py init_ui more edge cases

Browse Source
This commit is contained in:
wyzula-jan
2023-10-03 10:47:46 +02:00
parent 6c773c7c94
commit 36942b316a
2 changed files with 98 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bec_widgets/examples/extreme/extreme.py
View File

@ -183,8 +183,10 @@ class PlotApp(QWidget):
num_columns = num_plots num_columns = num_plots
self.plot_settings["num_columns"] = num_columns # Update the settings self.plot_settings["num_columns"] = num_columns # Update the settings
print( print(
f"Warning: num_columns in the YAML file was greater than the number of plots. Resetting num_columns to {num_columns}." f"Warning: num_columns in the YAML file was greater than the number of plots. Resetting num_columns to number of plots:{num_columns}."
) )
else:
self.plot_settings["num_columns"] = num_columns # Update the settings
num_rows = num_plots // num_columns num_rows = num_plots // num_columns
last_row_cols = num_plots % num_columns last_row_cols = num_plots % num_columns
@ -239,7 +241,7 @@ class PlotApp(QWidget):
curve_list = [] curve_list = []
for i, (y_config, color) in enumerate(zip(y_configs, colors_ys)): for i, (y_config, color) in enumerate(zip(y_configs, colors_ys)):
print(y_config) # print(y_config)
y_name = y_config["name"] y_name = y_config["name"]
y_entries = y_config.get("entry", [y_name]) y_entries = y_config.get("entry", [y_name])

137
tests/test_extreme.py
View File

@ -88,42 +88,6 @@ config_scan_mode = config = {
"scan_types": True, "scan_types": True,
}, },
"plot_data": { "plot_data": {
"line_scan": [
{
"plot_name": "BPM plot",
"x": {"label": "Motor X", "signals": [{"name": "samx"}]},
"y": {
"label": "BPM",
"signals": [
{"name": "gauss_bpm", "entry": "gauss_bpm"},
{"name": "gauss_adc1", "entry": "gauss_adc1"},
{"name": "gauss_adc2", "entry": "gauss_adc2"},
],
},
},
{
"plot_name": "Multi",
"x": {"label": "Motor X", "signals": [{"name": "samx", "entry": "samx"}]},
"y": {
"label": "Multi",
"signals": [
{"name": "gauss_bpm", "entry": "gauss_bpm"},
{"name": "samx", "entry": ["samx", "samx_setpoint"]},
],
},
},
{
"plot_name": "Multi",
"x": {"label": "Motor X", "signals": [{"name": "samx", "entry": "samx"}]},
"y": {
"label": "Multi",
"signals": [
{"name": "gauss_bpm", "entry": "gauss_bpm"},
{"name": "samx", "entry": ["samx", "samx_setpoint"]},
],
},
},
],
"grid_scan": [ "grid_scan": [
{ {
"plot_name": "Grid plot 1", "plot_name": "Grid plot 1",
@ -164,6 +128,42 @@ config_scan_mode = config = {
}, },
}, },
], ],
"line_scan": [
{
"plot_name": "BPM plot",
"x": {"label": "Motor X", "signals": [{"name": "samx"}]},
"y": {
"label": "BPM",
"signals": [
{"name": "gauss_bpm", "entry": "gauss_bpm"},
{"name": "gauss_adc1", "entry": "gauss_adc1"},
{"name": "gauss_adc2", "entry": "gauss_adc2"},
],
},
},
{
"plot_name": "Multi",
"x": {"label": "Motor X", "signals": [{"name": "samx", "entry": "samx"}]},
"y": {
"label": "Multi",
"signals": [
{"name": "gauss_bpm", "entry": "gauss_bpm"},
{"name": "samx", "entry": ["samx", "samx_setpoint"]},
],
},
},
{
"plot_name": "Multi",
"x": {"label": "Motor X", "signals": [{"name": "samx", "entry": "samx"}]},
"y": {
"label": "Multi",
"signals": [
{"name": "gauss_bpm", "entry": "gauss_bpm"},
{"name": "samx", "entry": ["samx", "samx_setpoint"]},
],
},
},
],
}, },
} }
@ -173,12 +173,7 @@ config_scan_mode = config = {
[ [
(config_device_mode_all_filled, "black", 2, "k"), (config_device_mode_all_filled, "black", 2, "k"),
(config_device_mode_no_entry, "white", 2, "w"), (config_device_mode_no_entry, "white", 2, "w"),
( (config_scan_mode, "white", 3, "w"),
config_scan_mode,
"white",
3,
"w",
),
], ],
) )
def test_init_config(qtbot, config, plot_setting_bg, num_plot, pg_background): def test_init_config(qtbot, config, plot_setting_bg, num_plot, pg_background):
@ -205,7 +200,63 @@ def test_init_config(qtbot, config, plot_setting_bg, num_plot, pg_background):
["BPM4i plots vs samy", "Gauss plots vs samx"], ["BPM4i plots vs samy", "Gauss plots vs samx"],
[(0, 0), (0, 1)], [(0, 0), (0, 1)],
), # num_columns greater than number of plots ), # num_columns greater than number of plots
# (config_scan_mode, 3, 3, ["BPM plot", "Multi", "Grid plot 1", "Grid plot 2", "Grid plot 3", "Grid plot 4"], [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2)]), (
config_device_mode_no_entry,
1,
1,
["BPM4i plots vs samy", "Gauss plots vs samx"],
[(0, 0), (1, 0)],
),
(
config_device_mode_no_entry,
2,
2,
["BPM4i plots vs samy", "Gauss plots vs samx"],
[(0, 0), (0, 1)],
),
(
config_device_mode_no_entry,
5,
2,
["BPM4i plots vs samy", "Gauss plots vs samx"],
[(0, 0), (0, 1)],
), # num_columns greater than number of plots,
(
config_scan_mode,
2,
2,
[
"Grid plot 1",
"Grid plot 2",
"Grid plot 3",
"Grid plot 4",
],
[(0, 0), (0, 1), (1, 0), (1, 1)],
),
(
config_scan_mode,
3,
3,
[
"Grid plot 1",
"Grid plot 2",
"Grid plot 3",
"Grid plot 4",
],
[(0, 0), (0, 1), (0, 2), (1, 0)],
),
(
config_scan_mode,
5,
4,
[
"Grid plot 1",
"Grid plot 2",
"Grid plot 3",
"Grid plot 4",
],
[(0, 0), (0, 1), (0, 2), (0, 3)],
), # num_columns greater than number of plots
], ],
) )
def test_init_ui( def test_init_ui(