def test_prime_queue_all_empty(): optimizer = BayesianOptimization(target_func, PBOUNDS, PTYPES, random_state=1) assert len(optimizer._queue) == 0 assert len(optimizer.space) == 0 optimizer._prime_queue(init_points=0) assert len(optimizer._queue) == 1 assert len(optimizer.space) == 0
def test_prime_queue_empty_with_init(): optimizer = BayesianOptimization(target_func, PBOUNDS, random_state=1) assert len(optimizer._queue) == 0 assert len(optimizer.space) == 0 optimizer._prime_queue(init_points=5) assert len(optimizer._queue) == 5 assert len(optimizer.space) == 0
def test_prime_queue_with_register_and_init(): optimizer = BayesianOptimization(target_func, PBOUNDS, PTYPES, random_state=1) assert len(optimizer._queue) == 0 assert len(optimizer.space) == 0 optimizer.register(params={"p1": 1, "p2": 2}, target=3) optimizer._prime_queue(init_points=3) assert len(optimizer._queue) == 3 assert len(optimizer.space) == 1