litellm-rs 0.4.16

A high-performance AI Gateway written in Rust, providing OpenAI-compatible APIs with intelligent routing, load balancing, and enterprise features
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

//! Tests for metrics module

#[cfg(test)]
use super::collector::MetricsCollector;
use super::helpers::{calculate_average, calculate_percentile};
use crate::config::models::monitoring::MonitoringConfig;
use std::collections::VecDeque;

#[test]
fn test_calculate_percentile() {
    let values = vec![1.0, 2.0, 3.0, 4.0, 5.0];
    assert_eq!(calculate_percentile(&values, 0.5), 3.0); // 50th percentile
    assert_eq!(calculate_percentile(&values, 0.95), 4.8); // 95th percentile (interpolated)
    assert_eq!(calculate_percentile(&values, 1.0), 5.0); // 100th percentile
    assert_eq!(calculate_percentile(&[], 0.5), 0.0); // empty array
}

#[test]
fn test_calculate_average() {
    let values: VecDeque<f64> = vec![1.0, 2.0, 3.0, 4.0, 5.0].into();
    assert_eq!(calculate_average(&values), 3.0);
    assert_eq!(calculate_average(&VecDeque::new()), 0.0);
}

#[tokio::test]
async fn test_metrics_collector_creation() {
    let config = MonitoringConfig::default();

    let collector = MetricsCollector::new(&config).await.unwrap();
    assert!(!collector.is_active());
}