nautilus_analysis/statistics/
loser_avg.rs1use std::fmt::Display;
17
18use nautilus_model::position::Position;
19
20use crate::{Returns, statistic::PortfolioStatistic};
21
22#[repr(C)]
27#[derive(Debug, Clone)]
28#[cfg_attr(
29 feature = "python",
30 pyo3::pyclass(module = "nautilus_trader.analysis", from_py_object)
31)]
32#[cfg_attr(
33 feature = "python",
34 pyo3_stub_gen::derive::gen_stub_pyclass(module = "nautilus_trader.analysis")
35)]
36pub struct AvgLoser {}
37
38impl Display for AvgLoser {
39 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
40 write!(f, "Avg Loser")
41 }
42}
43
44impl PortfolioStatistic for AvgLoser {
45 type Item = f64;
46
47 fn name(&self) -> String {
48 self.to_string()
49 }
50
51 fn calculate_from_realized_pnls(&self, realized_pnls: &[f64]) -> Option<Self::Item> {
52 if realized_pnls.is_empty() {
53 return Some(f64::NAN);
54 }
55
56 let losers: Vec<f64> = realized_pnls
57 .iter()
58 .filter(|&&pnl| pnl < 0.0)
59 .copied()
60 .collect();
61
62 if losers.is_empty() {
63 return Some(f64::NAN);
64 }
65
66 let sum: f64 = losers.iter().sum();
67 Some(sum / losers.len() as f64)
68 }
69
70 fn calculate_from_returns(&self, _returns: &Returns) -> Option<Self::Item> {
71 None
72 }
73
74 fn calculate_from_positions(&self, _positions: &[Position]) -> Option<Self::Item> {
75 None
76 }
77}
78
79#[cfg(test)]
80mod tests {
81 use nautilus_core::approx_eq;
82 use rstest::rstest;
83
84 use super::*;
85
86 #[rstest]
87 fn test_empty_pnls() {
88 let avg_loser = AvgLoser {};
89 let result = avg_loser.calculate_from_realized_pnls(&[]);
90 assert!(result.is_some());
91 assert!(result.unwrap().is_nan());
92 }
93
94 #[rstest]
95 fn test_no_losers() {
96 let avg_loser = AvgLoser {};
97 let pnls = vec![10.0, 20.0, 30.0];
98 let result = avg_loser.calculate_from_realized_pnls(&pnls);
99 assert!(result.is_some());
100 assert!(result.unwrap().is_nan());
101 }
102
103 #[rstest]
104 fn test_only_losers() {
105 let avg_loser = AvgLoser {};
106 let pnls = vec![-10.0, -20.0, -30.0];
107 let result = avg_loser.calculate_from_realized_pnls(&pnls);
108 assert!(result.is_some());
109 assert!(approx_eq!(f64, result.unwrap(), -20.0, epsilon = 1e-9));
110 }
111
112 #[rstest]
113 fn test_mixed_pnls() {
114 let avg_loser = AvgLoser {};
115 let pnls = vec![10.0, -20.0, 30.0, -40.0];
116 let result = avg_loser.calculate_from_realized_pnls(&pnls);
117 assert!(result.is_some());
118 assert!(approx_eq!(f64, result.unwrap(), -30.0, epsilon = 1e-9));
119 }
120
121 #[rstest]
122 fn test_zero_excluded() {
123 let avg_loser = AvgLoser {};
124 let pnls = vec![10.0, 0.0, -20.0, -30.0];
125 let result = avg_loser.calculate_from_realized_pnls(&pnls);
126 assert!(result.is_some());
127 assert!(approx_eq!(f64, result.unwrap(), -25.0, epsilon = 1e-9));
129 }
130
131 #[rstest]
132 fn test_single_loser() {
133 let avg_loser = AvgLoser {};
134 let pnls = vec![-10.0];
135 let result = avg_loser.calculate_from_realized_pnls(&pnls);
136 assert!(result.is_some());
137 assert!(approx_eq!(f64, result.unwrap(), -10.0, epsilon = 1e-9));
138 }
139
140 #[rstest]
141 fn test_name() {
142 let avg_loser = AvgLoser {};
143 assert_eq!(avg_loser.name(), "Avg Loser");
144 }
145}