mecha10-cli 0.1.47

Mecha10 CLI tool
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

// 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);
}