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)