Manuel Raimann
d81d1de4ff
- Delete 20 duplicate tests already covered by parameter_tests.rs - Move 11 pure Trial unit tests into src/trial.rs - Split remaining 84 integration tests into tests/study/ (9 modules) - Group sampler tests into tests/sampler/ (7 modules) - Group pruner tests into tests/pruner/ (2 modules)
44 lines
1.2 KiB
Rust
44 lines
1.2 KiB
Rust
use optimizer::parameter::{FloatParam, Parameter};
|
|
use optimizer::sampler::random::RandomSampler;
|
|
use optimizer::{Direction, Study};
|
|
|
|
#[test]
|
|
fn test_into_iterator_iterates_all_trials() {
|
|
let study: Study<f64> = Study::with_sampler(Direction::Minimize, RandomSampler::with_seed(42));
|
|
let x_param = FloatParam::new(0.0, 10.0);
|
|
|
|
for _ in 0..5 {
|
|
let mut trial = study.create_trial();
|
|
let x = x_param.suggest(&mut trial).unwrap();
|
|
study.complete_trial(trial, x * x);
|
|
}
|
|
|
|
let mut count = 0;
|
|
for trial in &study {
|
|
assert_eq!(trial.state, optimizer::TrialState::Complete);
|
|
count += 1;
|
|
}
|
|
assert_eq!(count, 5);
|
|
}
|
|
|
|
#[test]
|
|
fn test_into_iterator_empty_study() {
|
|
let study: Study<f64> = Study::new(Direction::Minimize);
|
|
|
|
let count = (&study).into_iter().count();
|
|
assert_eq!(count, 0);
|
|
}
|
|
|
|
#[test]
|
|
fn test_into_iterator_preserves_insertion_order() {
|
|
let study: Study<f64> = Study::new(Direction::Minimize);
|
|
|
|
for i in 0..3 {
|
|
let trial = study.create_trial();
|
|
study.complete_trial(trial, f64::from(i));
|
|
}
|
|
|
|
let ids: Vec<u64> = (&study).into_iter().map(|t| t.id).collect();
|
|
assert_eq!(ids, vec![0, 1, 2]);
|
|
}
|