def test_list_of_strings(experiment):
"""
Test saving list of strings via DataSaver
"""
p_values = ["X_Y", "X_X", "X_I", "I_I"]
list_of_strings = list(np.random.choice(p_values, (10, )))
p = qc.Parameter('p',
label='String parameter',
unit='',
get_cmd=None,
set_cmd=None,
initial_value='X_Y')
meas = Measurement(experiment)
meas.register_parameter(p, paramtype='text')
with meas.run() as datasaver:
datasaver.add_result((p, list_of_strings))
test_set = load_by_id(datasaver.run_id)
expec_data = np.array([item for item in list_of_strings])
actual_data = test_set.get_parameter_data()["p"]["p"]
try:
np.testing.assert_array_equal(actual_data, expec_data)
finally:
test_set.conn.close()
def test_string(experiment):
"""
Test that we can save text into database via Measurement API
"""
p = qc.Parameter('p', label='String parameter', unit='', get_cmd=None,
set_cmd=None, initial_value='some text')
meas = Measurement(experiment)
meas.register_parameter(p, paramtype='text')
with meas.run() as datasaver:
datasaver.add_result((p, "some text"))
test_set = load_by_id(datasaver.run_id)
assert test_set.get_data("p") == [["some text"]]
