use chrono::NaiveDate;
use yearfrac::DayCountConvention;

#[test]
fn test_accuracy() {
    let delta = 1e-9;

    let start = NaiveDate::from_ymd(1978, 2, 28);
    let end = NaiveDate::from_ymd(2020, 5, 17);
    let yf = DayCountConvention::from_int(0)
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 42.21388888889).abs() < delta);
    let yf = DayCountConvention::default().yearfrac(start, end);
    assert!((yf - 42.21388888889).abs() < delta);
    let yf = DayCountConvention::from_int(1)
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 42.21424933147).abs() < delta);
    let yf = DayCountConvention::from_int(2)
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 42.83055555556).abs() < delta);
    let yf = DayCountConvention::from_int(3)
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 42.24383561644).abs() < delta);
    let yf = DayCountConvention::from_int(4)
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 42.21944444444).abs() < delta);

    let start = NaiveDate::from_ymd(1993, 12, 2);
    let end = NaiveDate::from_ymd(2022, 4, 18);
    let yf = DayCountConvention::from_str("nasd30/360")
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 28.37777777778).abs() < delta);
    let yf = DayCountConvention::default().yearfrac(start, end);
    assert!((yf - 28.37777777778).abs() < delta);
    let yf = DayCountConvention::from_str("act/act")
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 28.37638039609).abs() < delta);
    let yf = DayCountConvention::from_str("act360")
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 28.78888888889).abs() < delta);
    let yf = DayCountConvention::from_str("act365")
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 28.39452054795).abs() < delta);
    let yf = DayCountConvention::from_str("eur30/360")
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 28.37777777778).abs() < delta);
}

#[test]
#[should_panic]
fn test_bad_input_str() {
    DayCountConvention::from_str("wrongvalue").unwrap();
}

/*
#[test]
#[should_panic]
fn test_bad_dates () {
    let start = NaiveDate::from_ymd(2032, 4, 28);
    let end = NaiveDate::from_ymd(2022, 5, 17);
    DayCountConvention::from_str("act/act").unwrap()
    .yearfrac(start, end);
}
*/

#[test]
fn test_start_after_end() {
    let delta = 1e-9;

    let end = NaiveDate::from_ymd(1978, 2, 28);
    let start = NaiveDate::from_ymd(2020, 5, 17);
    let yf = DayCountConvention::from_int(0)
        .unwrap()
        .yearfrac(start, end);
    assert!((yf - 42.21388888889).abs() < delta);
}

#[test]
fn test_yearfrac_signed() {
    let delta = 1e-9;

    let end = NaiveDate::from_ymd(1978, 2, 28);
    let start = NaiveDate::from_ymd(2020, 5, 17);
    let yf = DayCountConvention::from_int(0)
        .unwrap()
        .yearfrac_signed(start, end);
    assert!((yf + 42.21388888889).abs() < delta);
}