def test_gen_param_sweep():
ps = ParameterSweep()
f1 = ps.sweep.factors.add()
s1 = f1.sweeps.add()
s1.parameter_key = 'foo'
s1.points.points.extend([1, 2, 3])
f2 = ps.sweep.factors.add()
s2 = f2.sweeps.add()
s2.parameter_key = 'bar'
s2.points.points.extend([4, 5])
out = params.sweep_from_proto(ps)
assert out == Product(Zip(Points('foo', [1, 2, 3])),
Zip(Points('bar', [4, 5])))
def test_param_sweep_size():
ps = ParameterSweep()
f1 = ps.sweep.factors.add()
s11 = f1.sweeps.add()
s11.parameter_key = '11'
s11.linspace.num_points = 5
s12 = f1.sweeps.add()
s12.parameter_key = '12'
s12.points.points.extend(range(7))
f2 = ps.sweep.factors.add()
s21 = f2.sweeps.add()
s21.parameter_key = '21'
s21.linspace.num_points = 11
s22 = f2.sweeps.add()
s22.parameter_key = '22'
s22.points.points.extend(range(13))
assert len(params.sweep_from_proto(ps)) == 5 * 11
def test_param_sweep_keys():
ps = ParameterSweep()
f1 = ps.sweep.factors.add()
s11 = f1.sweeps.add()
s11.parameter_key = 'foo'
s11.points.points.extend(range(5))
s12 = f1.sweeps.add()
s12.parameter_key = 'bar'
s12.points.points.extend(range(7))
f2 = ps.sweep.factors.add()
s21 = f2.sweeps.add()
s21.parameter_key = 'baz'
s21.points.points.extend(range(11))
s22 = f2.sweeps.add()
s22.parameter_key = 'qux'
s22.points.points.extend(range(13))
out = params.sweep_from_proto(ps)
assert out.keys == ['foo', 'bar', 'baz', 'qux']
def test_empty_param_sweep_keys():
ps = ParameterSweep()
assert params.sweep_from_proto(ps).keys == []
def test_gen_empty_param_sweep():
ps = ParameterSweep()
out = params.sweep_from_proto(ps)
assert out == Unit
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 = ParameterSweep()
ps.sweep.factors.add()
ps.sweep.factors.add()
assert len(params.sweep_from_proto(ps)) == 0
def test_empty_param_sweep_size():
ps = ParameterSweep()
assert len(params.sweep_from_proto(ps)) == 1
