@@ -388,13 +388,14 @@ mod tests {
388
388
389
389
#[ test]
390
390
#[ serial]
391
- fn test_xsinx_wb2_egor_builder ( ) {
391
+ fn test_xsinx_trego_wb2_egor_builder ( ) {
392
392
let res = EgorBuilder :: optimize ( xsinx)
393
393
. configure ( |config| {
394
394
config
395
395
. max_iters ( 20 )
396
396
. regression_spec ( RegressionSpec :: ALL )
397
397
. correlation_spec ( CorrelationSpec :: ALL )
398
+ . trego ( true )
398
399
. seed ( 1 )
399
400
} )
400
401
. min_within ( & array ! [ [ 0.0 , 25.0 ] ] )
@@ -418,7 +419,7 @@ mod tests {
418
419
419
420
#[ test]
420
421
#[ serial]
421
- fn test_xsinx_checkpoint_egor ( ) {
422
+ fn test_xsinx_hot_start_egor ( ) {
422
423
let _ = std:: fs:: remove_file ( ".checkpoints/egor.arg" ) ;
423
424
let n_iter = 1 ;
424
425
let res = EgorBuilder :: optimize ( xsinx)
@@ -532,7 +533,7 @@ mod tests {
532
533
533
534
#[ test]
534
535
#[ serial]
535
- fn test_rosenbrock_2d_no_trego_egor_builder ( ) {
536
+ fn test_rosenbrock_2d_trego_egor_builder ( ) {
536
537
let outdir = "target/test_trego" ;
537
538
let _ = std:: fs:: remove_file ( format ! ( "{outdir}/{DOE_INITIAL_FILE}" ) ) ;
538
539
let _ = std:: fs:: remove_file ( format ! ( "{outdir}/{DOE_FILE}" ) ) ;
@@ -549,15 +550,16 @@ mod tests {
549
550
. max_iters ( max_iters)
550
551
. outdir ( outdir)
551
552
. seed ( 42 )
552
- . trego ( false )
553
+ . trego ( true )
553
554
} )
554
555
. min_within ( & xlimits)
555
556
. run ( )
556
557
. expect ( "Minimize failure" ) ;
557
558
let filepath = std:: path:: Path :: new ( & outdir) . join ( DOE_FILE ) ;
558
559
assert ! ( filepath. exists( ) ) ;
559
560
let doe: Array2 < f64 > = read_npy ( & filepath) . expect ( "file read" ) ;
560
- assert_eq ! ( doe. nrows( ) , init_doe. nrows( ) + max_iters) ; // we get one point per iter
561
+ assert ! ( doe. nrows( ) <= init_doe. nrows( ) + max_iters) ;
562
+ assert ! ( doe. nrows( ) >= init_doe. nrows( ) ) ;
561
563
562
564
println ! ( "Rosenbrock optim result = {res:?}" ) ;
563
565
println ! ( "Elapsed = {:?}" , now. elapsed( ) ) ;
0 commit comments