mecha10-behavior-patterns 0.1.23

Common behavior patterns for Mecha10 - subsumption, ensemble, and more
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

// Tests for mecha10_behavior_patterns::ensemble

use async_trait::async_trait;
use mecha10_behavior_patterns::*;
use mecha10_behavior_runtime::prelude::*;

#[derive(Debug)]
struct MockModel {
    status: NodeStatus,
}

impl MockModel {
    fn new(status: NodeStatus) -> Self {
        Self { status }
    }
}

#[async_trait]
impl BehaviorNode for MockModel {
    async fn tick(&mut self, _ctx: &Context) -> anyhow::Result<NodeStatus> {
        Ok(self.status)
    }
}

#[tokio::test]
#[ignore = "Requires Redis infrastructure"]
async fn test_ensemble_conservative_all_succeed() {
    let ctx = Context::new("test").await.unwrap();

    let mut ensemble = EnsembleNode::new(EnsembleStrategy::Conservative)
        .add_model(Box::new(MockModel::new(NodeStatus::Success)), 1.0)
        .add_model(Box::new(MockModel::new(NodeStatus::Success)), 1.0);

    let status = ensemble.tick(&ctx).await.unwrap();
    assert_eq!(status, NodeStatus::Success);
}

#[tokio::test]
#[ignore = "Requires Redis infrastructure"]
async fn test_ensemble_conservative_one_fails() {
    let ctx = Context::new("test").await.unwrap();

    let mut ensemble = EnsembleNode::new(EnsembleStrategy::Conservative)
        .add_model(Box::new(MockModel::new(NodeStatus::Success)), 1.0)
        .add_model(Box::new(MockModel::new(NodeStatus::Failure)), 1.0);

    let status = ensemble.tick(&ctx).await.unwrap();
    assert_eq!(status, NodeStatus::Failure);
}

#[tokio::test]
#[ignore = "Requires Redis infrastructure"]
async fn test_ensemble_optimistic_any_succeeds() {
    let ctx = Context::new("test").await.unwrap();

    let mut ensemble = EnsembleNode::new(EnsembleStrategy::Optimistic)
        .add_model(Box::new(MockModel::new(NodeStatus::Success)), 1.0)
        .add_model(Box::new(MockModel::new(NodeStatus::Failure)), 1.0);

    let status = ensemble.tick(&ctx).await.unwrap();
    assert_eq!(status, NodeStatus::Success);
}

#[tokio::test]
#[ignore = "Requires Redis infrastructure"]
async fn test_ensemble_majority() {
    let ctx = Context::new("test").await.unwrap();

    let mut ensemble = EnsembleNode::new(EnsembleStrategy::Majority)
        .add_model(Box::new(MockModel::new(NodeStatus::Success)), 1.0)
        .add_model(Box::new(MockModel::new(NodeStatus::Success)), 1.0)
        .add_model(Box::new(MockModel::new(NodeStatus::Failure)), 1.0);

    let status = ensemble.tick(&ctx).await.unwrap();
    assert_eq!(status, NodeStatus::Success); // 2 out of 3 succeed
}

#[tokio::test]
#[ignore = "Requires Redis infrastructure"]
async fn test_ensemble_weighted_vote() {
    let ctx = Context::new("test").await.unwrap();

    let mut ensemble = EnsembleNode::new(EnsembleStrategy::WeightedVote)
        .with_threshold(0.6)
        .add_model(Box::new(MockModel::new(NodeStatus::Success)), 0.7) // 70% weight succeeds
        .add_model(Box::new(MockModel::new(NodeStatus::Failure)), 0.3); // 30% weight fails

    let status = ensemble.tick(&ctx).await.unwrap();
    assert_eq!(status, NodeStatus::Success); // 0.7 >= 0.6 threshold
}