swarm-engine-eval 0.1.6

Evaluation framework for SwarmEngine
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

//! JSON reporter

use crate::error::Result;
use crate::reporter::{EvalReport, Reporter};

/// JSON reporter
pub struct JsonReporter {
    /// Pretty print
    pretty: bool,
}

impl JsonReporter {
    /// Create new JSON reporter
    pub fn new() -> Self {
        Self { pretty: true }
    }

    /// Set pretty printing
    pub fn with_pretty(mut self, pretty: bool) -> Self {
        self.pretty = pretty;
        self
    }
}

impl Default for JsonReporter {
    fn default() -> Self {
        Self::new()
    }
}

impl Reporter for JsonReporter {
    fn generate(&self, report: &EvalReport) -> Result<String> {
        let json = if self.pretty {
            serde_json::to_string_pretty(report)?
        } else {
            serde_json::to_string(report)?
        };
        Ok(json)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::aggregator::AggregatedResults;
    use crate::reporter::{ConfigSummary, SeedInfo};

    #[test]
    fn test_json_reporter() {
        let report = EvalReport {
            config_summary: ConfigSummary {
                scenario_name: "test".to_string(),
                scenario_id: "test:v1".to_string(),
                worker_count: 2,
                max_ticks: 100,
                run_count: 10,
            },
            seed_info: SeedInfo {
                base_seed: 12345,
                run_seeds: vec![12345],
            },
            runs: vec![],
            aggregated: AggregatedResults::default(),
            assertion_results: vec![],
        };

        let reporter = JsonReporter::new();
        let json = reporter.generate(&report).unwrap();
        assert!(json.contains("config_summary"));
        assert!(json.contains("aggregated"));
    }
}