def test_save():
input_variables = {
"input1": ScalarInputVariable(name="input1", default=1, range=[0.0, 5.0]),
"input2": ScalarInputVariable(name="input2", default=2, range=[0.0, 5.0]),
}
output_variables = {
"output1": ScalarOutputVariable(name="output1"),
"output2": ScalarOutputVariable(name="output2"),
}
file_name = "test_variables.pickle"
utils.save_variables(input_variables, output_variables, file_name)
utils.load_variables(file_name)
os.remove(file_name)
from bokeh.layouts import column, row, Spacer
from bokeh.models import Div
from bokeh.models.widgets import Select
from bokeh import palettes
from bokeh.io import curdoc
#from bokeh.resources import INLINE#
from lume_model.utils import load_variables
from lume_epics.client.widgets.plots import ImagePlot, Striptool
from lume_epics.client.widgets.controls import build_sliders
from lume_epics.client.widgets.tables import ValueTable
from lume_epics.client.controller import Controller
prefix = "test" # Prefix used by our server
variable_filename = "files/surrogate_model_variables_2.pickle"
input_variables, output_variables = load_variables(variable_filename)
controller = Controller("ca")
inputs = [
input_variables["distgen:r_dist:sigma_xy:value"],
input_variables["distgen:t_dist:length:value"],
# input_variables["distgen:total_charge:value"],
input_variables["SOL1:solenoid_field_scale"],
input_variables["CQ01:b1_gradient"],
input_variables["SQ01:b1_gradient"],
input_variables["L0A_phase:dtheta0_deg"],
# input_variables["L0A_scale:voltage"],
input_variables["end_mean_z"]
]