From d0aa393e0b47fc0aed0d40643cfe6cfd787e8f1a Mon Sep 17 00:00:00 2001 From: David Hewson Date: Mon, 27 Feb 2023 13:20:13 +0000 Subject: [PATCH] change tests to assert whole year at once --- src/uk.rs | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/src/uk.rs b/src/uk.rs index 0dcccbf..1e4ab68 100644 --- a/src/uk.rs +++ b/src/uk.rs @@ -85,16 +85,19 @@ mod tests { } else { 365 }; - for i in 0..days { - let date = jan1 + Duration::days(i); - let holiday = date.is_bankholiday(); - let expected = $dates.contains(&(date.day(), date.month())); - assert!(expected == holiday, - format!("Expected {} for {} but got {}", expected, date, holiday)); - assert_eq!(is_bankholiday(&date), holiday); - } + let bhs: Vec<_> = (0..days) + .map(|i| jan1 + Duration::days(i)) + .filter_map(|date| { + if date.is_bankholiday() { + Some((date.day(), date.month())) + } else { + None + } + }) + .collect(); + assert_eq!($dates, bhs.as_slice()); } - } + }; } test!(year_1999, 1999,