def test_2d_nd_multi_measurement():
m, n, o = 5, 7, 3
param_x = lambda s, n, v: {
"x": {
"unit": "",
"value": v,
"independent_parameter": True
}
}
coordinate_values_x = np.arange(m)
param_y = lambda s, n, v: {
"y": {
"unit": "",
"value": v,
"independent_parameter": True
}
}
coordinate_values_y = np.arange(n)
measure_values1 = np.arange(m * n * o).reshape((m * n, o))
g = (i for i in measure_values1)
measure1 = lambda s, n: {"m": {"unit": "", "value": next(g)}}
measure_values2 = np.linspace(-1, 1, m)
h = (i for i in measure_values2)
measure2 = lambda s, n: {"n": {"unit": "", "value": next(h)}}
so = sweep(param_x,
coordinate_values_x)(measure2,
sweep(param_y,
coordinate_values_y)(measure1))
store = NpStorage()
for i in so:
store.add(i)
assert np.all(store["m"]["m []"] == measure_values1)
tmpx = store["m"]["x []"].reshape((m, n))
tmpy = store["m"]["y []"].reshape((m, n))
assert np.all(
np.vstack([tmpy, tmpx]) == np.vstack(
np.meshgrid(coordinate_values_y, coordinate_values_x)))
assert np.all(store["n"]["n []"].flatten() == measure_values2)
assert np.all(store["n"]["x []"].flatten() == coordinate_values_x)
assert "y []" not in store["n"]
def test_shape_mismatch_exception():
param = lambda s, n, v: {
"x": {
"unit": "",
"value": v,
"independent_parameter": True
}
}
measure_values = [[0, 1, 2], [3, 4, 5], [6, 7, 8, 9],
[10, 11, 12]] # The third one is longer on purpose
g = (i for i in measure_values)
measure = lambda s, n: {"m": {"unit": "", "value": next(g)}}
coordinate_values = np.array([0, 1, 2, 3])
so = sweep(param, coordinate_values)(measure)
store = NpStorage()
for count, i in enumerate(so):
if count == 2:
with pytest.raises(TypeError) as e:
store.add(i)
assert str(e.value) == "The shape or dtype of the measurement and/or independent parameter has suddenly " \
"changed. This is not allowed"
else:
store.add(i)
def test_data_set():
xvals = np.linspace(0, 1, 10)
yvals = np.sin(xvals)
param = lambda s, n, v: {
"x": {
"unit": "",
"value": v,
"independent_parameter": True
}
}
measure = lambda s, n, v: {"m": {"unit": "", "value": v}}
dset = param(None, None, xvals[:5])
dset.update(measure(None, None, yvals[:5]))
store = NpStorage()
store.dataset(dset)
g = iter(yvals[5:])
measure2 = lambda s, n: measure(s, n, next(g))
so = sweep(param, xvals[5:])(measure2)
for i in so:
store.add(i)
assert np.all(store["m"]["m []"].flatten() == yvals)
assert np.all(store["m"]["x []"].flatten() == xvals)
def test_delay():
N = 16
param = lambda s, n, v: {
"x": {
"unit": "",
"value": v,
"independent_parameter": True
}
}
measure_values = np.linspace(-1, 1, N)
def measure(s, n):
n.count += 1
c = n.count
if c % 4 == 0:
rval = DataSet(measure_values[c - 4:c])
else:
rval = Delayed()
return {"m": {"unit": "", "value": rval}}
namespace = Namespace()
namespace.count = 0
coordinate_values = np.array(range(N))
so = sweep(param, coordinate_values)(measure).set_namespace(namespace)
store = NpStorage()
for i in so:
store.add(i)
assert np.all(store["m"]["m []"].flatten() == measure_values)
assert np.all(store["m"]["x []"].flatten() == coordinate_values)
def test():
param = lambda s, n, v: {
"x": {
"unit": "V",
"value": v,
"independent_parameter": True
}
}
measure_values = np.linspace(-1, 1, 4)
g = (i for i in measure_values)
measure = lambda s, n: {"m": {"unit": "A", "value": next(g)}}
coordinate_values = np.array([0, 1, 2, 3])
so = sweep(param, coordinate_values)(measure)
store = WebUIWriter("")
for i in so:
store.add(i)
pages = store.get_pages()
json_dict = store.page_to_json(pages["m"])
assert json_dict["type"] == "linear"
assert np.all(json_dict["x"]["data"] == coordinate_values)
assert json_dict["x"]["unit"] == "V"
assert np.all(json_dict["y"]["data"] == measure_values)
assert json_dict["y"]["unit"] == "A"
def test2(params, values, stdio, measure, namespace):
p1, p2 = params[:2]
setup, cleanup, measure_main = measure[:3]
sweep_values = values[0]
measurement = Measurement(
setup,
cleanup,
(
sweep(p1, sweep_values),
[
measure_main,
p2
]
)
).run()
out1 = measurement.output(p2.label)
assert out1[p1.label] == {"unit": p1.unit, "value": list(sweep_values), "independent_parameter": True}
assert out1[p2.label] == {"unit": p2.unit, "value": len(sweep_values) * [0]}
assert measure_main.name not in out1
out2 = measurement.output(measure_main.name)
assert out2[p1.label] == {"unit": p1.unit, "value": list(sweep_values), "independent_parameter": True}
assert measure_main.name in out2
return str(stdio)
def test_2d():
m, n = 5, 7
param_x = lambda s, n, v: {
"x": {
"unit": "",
"value": v,
"independent_parameter": True
}
}
coordinate_values_x = np.arange(m)
param_y = lambda s, n, v: {
"y": {
"unit": "",
"value": v,
"independent_parameter": True
}
}
coordinate_values_y = np.arange(n)
measure_values = np.linspace(-1, 1, m * n)
g = (i for i in measure_values)
measure = lambda s, n: {"m": {"unit": "", "value": next(g)}}
so = sweep(param_x,
coordinate_values_x)(sweep(param_y,
coordinate_values_y)(measure))
store = NpStorage()
for i in so:
store.add(i)
assert np.all(store["m"]["m []"].flatten() == measure_values)
tmpx = store["m"]["x []"].reshape((m, n))
tmpy = store["m"]["y []"].reshape((m, n))
assert np.all(
np.vstack([tmpy, tmpx]) == np.vstack(
np.meshgrid(coordinate_values_y, coordinate_values_x)))
def test(params, values, stdio, measure, namespace):
p = params[0]
setup, cleanup = measure[:2]
sweep_values = values[0]
measurement = Measurement(
setup,
cleanup,
sweep(p, sweep_values)
).run()
out = measurement.output(p.label)
assert out == {p.label: {"unit": p.unit, "value": list(sweep_values), "independent_parameter": True}}
return str(stdio)
def test(params, values, stdio, measure, namespace):
measurement_function = hardware_measurement([0, 1, 2, 3])
p = params[0]
soft_sweep_values = [9, 10, 11]
so = sweep(p, soft_sweep_values)(hardware_sweep(measurement_function))
setup, cleanup = measure[:2]
measurement = Measurement(
setup,
cleanup,
so
).run()
out = measurement.output("measurement")
print(out)
def test_preserve_param_order():
"""
Test that the order in which the independent parameter names appear in the store dtype is in the order whereby
the inner most loop appears first, the first outer loop appears second, etc.
"""
n_params = 6
axis_lengths = n_params * [3]
alphabet = [chr(i) for i in range(ord("a"), ord("z") + 1)]
names = np.random.choice(alphabet, n_params, replace=False)
def param_function(name):
def inner(s, n, v):
return {
name: {
"unit": "",
"value": v,
"independent_parameter": True
}
}
return inner
params = [param_function(name) for name in names]
coordinate_values = [range(r) for r in axis_lengths]
measure_values = np.linspace(-1, 1, int(np.prod(axis_lengths)))
g = (i for i in measure_values)
measure = lambda s, n: {"measure": {"unit": "", "value": next(g)}}
so = measure
for p, c in zip(params, coordinate_values):
so = sweep(p, c)(so)
store = NpStorage()
for i in so:
store.add(i)
dtype_names = store["measure"].dtype.names
assert all([
"{} []".format(name) == dtype_name
for (name, dtype_name) in zip(names, dtype_names)
])
def test(params, values, stdio, measure, namespace):
p = params[0]
setup, cleanup, measure_main = measure[:3]
sweep_values = values[0]
measurement = Measurement(
setup,
cleanup,
(
sweep(p, sweep_values),
measure_main
)
).run()
out = measurement.output(p.label)
assert out[p.label] == {"unit": p.unit, "value": list(sweep_values), "independent_parameter": True}
assert measure_main.name in out
return str(stdio)
def test_1d():
param = lambda s, n, v: {
"x": {
"unit": "",
"value": v,
"independent_parameter": True
}
}
measure_values = np.linspace(-1, 1, 4)
g = (i for i in measure_values)
measure = lambda s, n: {"m": {"unit": "", "value": next(g)}}
coordinate_values = np.array([0, 1, 2, 3])
so = sweep(param, coordinate_values)(measure)
store = NpStorage()
for i in so:
store.add(i)
assert np.all(store["m"]["m []"].flatten() == measure_values)
assert np.all(store["m"]["x []"].flatten() == coordinate_values)
