use crate::calendar::Calendar;
use crate::utilities::unpack_date;
use time::{Date, Month, Weekday};
use RustQuant_iso::*;
pub struct CanadaCalendar;
impl Calendar for CanadaCalendar {
fn new() -> Self {
Self
}
fn name(&self) -> &'static str {
"Canada"
}
fn country_code(&self) -> ISO_3166 {
CANADA
}
fn market_identifier_code(&self) -> ISO_10383 {
XCNQ
}
fn is_holiday(&self, date: Date) -> bool {
let (y, m, d, wd, yd, em) = unpack_date(date, false);
if (
((d == 1 || ((d == 2 || d == 3) && wd == Weekday::Monday)) && m == Month::January)
|| ((15..=21).contains(&d) && wd == Weekday::Monday && m == Month::February && y >= 2008)
|| (yd == em-3)
|| (d > 17 && d <= 24 && wd == Weekday::Monday && m == Month::May)
|| ((d == 1 || ((d == 2 || d == 3) && wd == Weekday::Monday)) && m==Month::July)
|| (d <= 7 && wd == Weekday::Monday && m == Month::August)
|| (d <= 7 && wd == Weekday::Monday && m == Month::September)
|| (((d == 30 && m == Month::September) || (d <= 2 && m == Month::October && wd == Weekday::Monday)) && y >= 2021)
|| (d > 7 && d <= 14 && wd == Weekday::Monday && m == Month::October)
|| ((d == 11 || ((d == 12 || d == 13) && wd == Weekday::Monday)) && m == Month::November)
|| ((d == 25 || (d == 27 && (wd == Weekday::Monday || wd == Weekday::Tuesday))) && m == Month::December)
|| ((d == 26 || (d == 28 && (wd == Weekday::Monday || wd == Weekday::Tuesday))) && m == Month::December)
) {
return true;
}
false
}
}
#[cfg(test)]
mod test_canada {
use super::*;
use time::macros::date;
#[test]
fn test_name() {
let calendar = CanadaCalendar;
assert_eq!(calendar.name(), "Canada");
}
#[test]
fn test_is_weekend() {
let calendar = CanadaCalendar;
let sat = date!(2023 - 08 - 26);
let sun = date!(2023 - 08 - 27);
assert!(!calendar.is_business_day(sat));
assert!(!calendar.is_business_day(sun));
}
#[test]
fn test_is_public_holiday() {
let calendar = CanadaCalendar;
let new_years_day = date!(2023 - 01 - 01);
let family_day = date!(2023 - 02 - 20); let canada_day = date!(2023 - 07 - 01);
let thanksgiving = date!(2023 - 10 - 09); let christmas = date!(2023 - 12 - 25);
assert!(!calendar.is_business_day(new_years_day));
assert!(!calendar.is_business_day(family_day));
assert!(!calendar.is_business_day(canada_day));
assert!(!calendar.is_business_day(thanksgiving));
assert!(!calendar.is_business_day(christmas));
}
#[test]
fn test_is_regular_business_day() {
let calendar = CanadaCalendar;
let regular_day1 = date!(2023 - 03 - 01);
let regular_day2 = date!(2023 - 07 - 12);
let regular_day3 = date!(2023 - 11 - 17);
assert!(calendar.is_business_day(regular_day1));
assert!(calendar.is_business_day(regular_day2));
assert!(calendar.is_business_day(regular_day3));
}
}