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"]]