def test_len():
exp = Experiment("TestLen")
exp.add_params(p1=1, p2=[2, 3], p3="param")
exp.add_params(p1=4, p2=5)
assert_equal(len(exp), 6)
i = 0
for _ in exp:
i += 1
assert_equal(i, 6)
assert_equal(len(exp), 6)
def test_exp_diff():
exp = Experiment("TestExpDiff")
exp.add_params(p1=1, p2=[2, 3], p3="param")
exp.add_params(p1=4, p2=5)
labels = []
params = []
for l, p in exp:
labels.append(l)
params.append(p)
for i in range(len(labels)):
lab = labels[:i]
par = params[:i]
cmpts = {k:v for k,v in zip(lab, par)}
res = experiment_diff(exp, cmpts)
for l, p in res:
assert_in(l, labels[i:])
assert_in(p, params[i:])
def test_getitem():
exp = Experiment("TestItem")
exp.add_params(p1=1, p2=[3, 2], p3="param")
exp.add_params(p1=4, p2=5)
assert_raises(KeyError, exp.get_params_for, -1)
assert_raises(KeyError, exp.get_params_for, 10e6)
labels = []
params = []
for l, p in exp:
labels.append(l)
params.append(p)
for i, (l, p) in enumerate(zip(labels, params)):
li, pi = exp[l]
assert_equal(l, li)
assert_equal(p, pi)
li, pi = exp[i]
assert_equal(l, li)
assert_equal(p, pi)
