// Tests for mecha10_cli::handlers::train

use mecha10_cli::handlers::train::*;

#[test]
fn test_train_args() {
    let args = TrainArgs {
        config: PathBuf::from("rl/configs/ppo_navigation.yaml"),
        wait: true,
    };

    assert_eq!(args.config, PathBuf::from("rl/configs/ppo_navigation.yaml"));
    assert!(args.wait);
}

#[test]
fn test_train_args_no_wait() {
    let args = TrainArgs {
        config: PathBuf::from("rl/configs/sac_manipulation.yaml"),
        wait: false,
    };

    assert!(!args.wait);
}
