def test_sweep_from_proto_missing_type():
ps = params_pb2.ParameterSweep(sweep=params_pb2.ProductSweep(factors=[
params_pb2.ZipSweep(
sweeps=[params_pb2.SingleSweep(parameter_key='foo')])
]))
with pytest.raises(ValueError):
params.sweep_from_proto(ps)
def test_param_sweep_keys():
ps = params_pb2.ParameterSweep(
sweep=params_pb2.ProductSweep(
factors=[
params_pb2.ZipSweep(
sweeps=[
params_pb2.SingleSweep(
parameter_key='foo', points=params_pb2.Points(points=range(5))
),
params_pb2.SingleSweep(
parameter_key='bar', points=params_pb2.Points(points=range(7))
),
]
),
params_pb2.ZipSweep(
sweeps=[
params_pb2.SingleSweep(
parameter_key='baz', points=params_pb2.Points(points=range(11))
),
params_pb2.SingleSweep(
parameter_key='qux', points=params_pb2.Points(points=range(13))
),
]
),
]
)
)
out = params.sweep_from_proto(ps)
assert out.keys == ['foo', 'bar', 'baz', 'qux']
def test_gen_param_sweep():
ps = params_pb2.ParameterSweep(
sweep=params_pb2.ProductSweep(
factors=[
params_pb2.ZipSweep(
sweeps=[
params_pb2.SingleSweep(
parameter_key='foo', points=params_pb2.Points(points=[1, 2, 3])
)
]
),
params_pb2.ZipSweep(
sweeps=[
params_pb2.SingleSweep(
parameter_key='bar', points=params_pb2.Points(points=[4, 5])
)
]
),
]
)
)
out = params.sweep_from_proto(ps)
assert out == cirq.Product(
cirq.Zip(cirq.Points('foo', [1, 2, 3])), cirq.Zip(cirq.Points('bar', [4, 5]))
)
def test_param_sweep_size():
ps = params_pb2.ParameterSweep(sweep=params_pb2.ProductSweep(factors=[
params_pb2.ZipSweep(sweeps=[
params_pb2.SingleSweep(parameter_key='11',
linspace=params_pb2.Linspace(first_point=0,
last_point=10,
num_points=5)),
params_pb2.SingleSweep(parameter_key='12',
points=params_pb2.Points(points=range(7)))
]),
params_pb2.ZipSweep(sweeps=[
params_pb2.SingleSweep(
parameter_key='21',
linspace=params_pb2.Linspace(
first_point=0, last_point=10, num_points=11)),
params_pb2.SingleSweep(parameter_key='22',
points=params_pb2.Points(points=range(13)))
])
]))
# Sweeps sx1 and sx2 are zipped, so should use num number of points.
# These are then producted, so this should multiply number of points.
assert len(params.sweep_from_proto(ps)) == 5 * 11
def test_empty_param_sweep_keys():
assert params.sweep_from_proto(params_pb2.ParameterSweep()).keys == []
def test_gen_empty_param_sweep():
out = params.sweep_from_proto(params_pb2.ParameterSweep())
assert out == cirq.UnitSweep
def test_sweep_to_proto(sweep, expected):
proto = params.sweep_to_proto(sweep)
out = params.sweep_from_proto(proto)
assert out == expected
def test_param_sweep_size_versus_gen(param_sweep):
sweep = params.sweep_from_proto(param_sweep)
predicted_size = len(sweep)
out = list(sweep)
assert len(out) == predicted_size
def test_param_sweep_size_no_sweeps():
ps = params_pb2.ParameterSweep(sweep=params_pb2.ProductSweep(
factors=[params_pb2.ZipSweep(),
params_pb2.ZipSweep()]))
assert len(params.sweep_from_proto(ps)) == 1
def test_empty_param_sweep_size():
assert len(params.sweep_from_proto(params_pb2.ParameterSweep())) == 1
