wbt 0.1.6

Weight-based backtesting engine for quantitative trading
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

use polars::error::PolarsError;
use thiserror::Error;

#[derive(Debug, Error)]
pub enum WbtError {
    #[error("expected value for {0}, got None")]
    NoneValue(String),

    #[error("io: {0}")]
    Io(String),

    #[error("polars: {0}")]
    Polars(#[from] PolarsError),

    #[error("returns should not be empty")]
    ReturnsEmpty,

    #[error("{0:#}")]
    Unexpected(#[from] anyhow::Error),
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn none_value_display() {
        let err = WbtError::NoneValue("test_field".into());
        assert_eq!(err.to_string(), "expected value for test_field, got None");
    }

    #[test]
    fn returns_empty_display() {
        let err = WbtError::ReturnsEmpty;
        assert_eq!(err.to_string(), "returns should not be empty");
    }

    #[test]
    fn from_anyhow() {
        let anyhow_err = anyhow::anyhow!("something went wrong");
        let wbt_err: WbtError = anyhow_err.into();
        assert!(wbt_err.to_string().contains("something went wrong"));
    }
}