mecha10-cli 0.8.0

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
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65


use mecha10_cli::sim::robot_generator::*;

#[test]
fn test_basic_robot_generation() {
}

#[test]
fn test_robot_dimensions() {
    let profile = RobotProfile {
        platform: "differential-drive".to_string(),
        sensors: vec!["camera".to_string(), "lidar".to_string()],
        sensor_configs: vec![],
        task_nodes: vec![],
        capabilities: vec![],
        difficulty_preference: "medium".to_string(),
    };

    let generator = RobotGenerator { profile, rl_mode: true };

    assert_eq!(generator.get_robot_radius(), 0.3);
    assert_eq!(generator.get_robot_height(), 1.0);
}

#[test]
fn test_tscn_header() {
    let profile = RobotProfile {
        platform: "differential-drive".to_string(),
        sensors: vec![],
        sensor_configs: vec![],
        task_nodes: vec![],
        capabilities: vec![],
        difficulty_preference: "medium".to_string(),
    };

    let generator = RobotGenerator {
        profile,
        rl_mode: false,
    };
    let tscn = generator.generate_tscn();

    assert!(tscn.contains("[gd_scene load_steps=3 format=3]"));
    assert!(tscn.contains("[node name=\"Robot\" type=\"CharacterBody3D\"]"));
}

#[test]
fn test_rl_mode_includes_ai_controller() {
    let profile = RobotProfile {
        platform: "differential-drive".to_string(),
        sensors: vec!["camera".to_string()],
        sensor_configs: vec![],
        task_nodes: vec![],
        capabilities: vec![],
        difficulty_preference: "medium".to_string(),
    };

    let generator = RobotGenerator { profile, rl_mode: true };
    let tscn = generator.generate_tscn();

    assert!(tscn.contains("[gd_scene load_steps=5 format=3]"));
    assert!(tscn.contains("[node name=\"AIController3D\""));
    assert!(tscn.contains("rover_base_rl.gd"));
    assert!(tscn.contains("rover_ai_controller.gd"));
    assert!(tscn.contains("[node name=\"CameraRL\""));
}