test_rosenbrock.py 文件源码

def test_rosenbrock():
    initial_state = [tuple((random.normalvariate(0, 5) for _ in xrange(DIMENSION)))
                     for x in xrange(N_ANNEALERS)]

    annealer = CoupledAnnealer(
        rosenbrock,
        probe,
        n_annealers=N_ANNEALERS,
        initial_state=initial_state,
        steps=STEPS,
        processes=4,
        verbose=1,
    )

    annealer.anneal()
    energy, state = annealer.get_best()

    assert len(state) == DIMENSION